import torch

print(torch.cuda.is_available())

print(torch.__version__)