PyTorch torch.cuda.is_available función
Pytorch torch manual de referencia
torch.cuda.is_availableEs una función en PyTorch que se utiliza para comprobar si CUDA está disponible.
Definición de la función
torch.cuda.is_available()
Ejemplo de uso
Ejemplo
import torch
# Comprobar si CUDA está disponible
if torch.cuda.is_available():
print("CUDA disponible!")
print("Número de GPU:", torch.cuda.device_count())
print("Nombre de la GPU:", torch.cuda.get_device_name(0))
else:
print("CUDA no disponible")
# Comprobar si CUDA está disponible
if torch.cuda.is_available():
print("CUDA disponible!")
print("Número de GPU:", torch.cuda.device_count())
print("Nombre de la GPU:", torch.cuda.get_device_name(0))
else:
print("CUDA no disponible")
Otras extensiones