import torch print(torch.cuda.is_available())