035df3d 36ed17a 035df3d
1
2
3
4
5
6
import torch print(torch.cuda.is_available()) # Should return True if our GPU is enabled print(torch.cuda.get_device_name(0)) # Should return "Tesla T4" if our GPU is enabled print(torch.version.cuda) # Should return "12.4" if our GPU is enabled
