Fonction torch.is_available de PyTorch
Manuel de référence PyTorch torch
torch.is_availableest une fonction de PyTorch utilisée pour vérifier si CUDA est disponible. Renvoie True si CUDA est disponible, False sinon.
Définition de fonction
torch.is_available()
Exemple d'utilisation
Exemple
import torch
# Vérifier si CUDA est disponible
if torch.is_available():
print("CUDA est disponible !")
print(f"Nombre de périphériques CUDA : {torch.cuda.device_count()}")
else:
print("CUDA n'est pas disponible")
# Vérifier si CUDA est disponible
if torch.is_available():
print("CUDA est disponible !")
print(f"Nombre de périphériques CUDA : {torch.cuda.device_count()}")
else:
print("CUDA n'est pas disponible")
Autres extensions