torch.cuda.is_available() torch.cuda.current_device() torch.cuda.device(0) torch.cuda.device_count() torch.cuda.get_device_name(0)
print(torch.version.cuda)