The GPu should be enabled

print(“torch.cuda.is_available()   =”, torch.cuda.is_available())
print(“torch.cuda.device_count()   =”, torch.cuda.device_count())
print(“torch.cuda.device(‘cuda’)   =”, torch.cuda.device(‘cuda’))
print(“torch.cuda.current_device() =”, torch.cuda.current_device())
torch.cuda.get_device_name(0)