import torch
print(torch.__version__)
print(torch.cuda.device_count())
print(torch.version.cuda)
print(torch.cuda.is_available())
import torch
print(torch.__version__)
print(torch.cuda.device_count())
print(torch.version.cuda)
print(torch.cuda.is_available())