import torch
print(torch.__version__)
print("CUDA Available:", torch.cuda.is_available())