File size: 228 Bytes
22d8ab7 |
1 2 3 4 5 6 7 8 9 10 11 |
import torch
# Verifica si CUDA está disponible
print(torch.cuda.is_available())
# Muestra la versión de CUDA que está utilizando PyTorch
print(torch.version.cuda)
# Muestra la versión de PyTorch
print(torch.__version__)
|