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__)