Función PyTorch torch.is_available
Manual de referencia de Pytorch torch
torch.is_availableEs una función en PyTorch utilizada para comprobar si CUDA está disponible. Devuelve True si CUDA está disponible, False si no lo está.
Definición de la función
torch.is_available()
Ejemplo de uso
Ejemplo
import torch
# Comprueba si CUDA está disponible
if torch.is_available():
print("CUDA is available!")
print(f"CUDA device count: {torch.cuda.device_count()}")
else:
print("CUDA is not available")
# Comprueba si CUDA está disponible
if torch.is_available():
print("CUDA is available!")
print(f"CUDA device count: {torch.cuda.device_count()}")
else:
print("CUDA is not available")
Otras extensiones