File size: 238 Bytes
2d984ae |
1 2 3 4 5 6 |
import torch, platform, sys
print("Torch:", torch.__version__)
print("Built for CUDA:", torch.version.cuda)
print("CUDA available:", torch.cuda.is_available())
if torch.cuda.is_available():
print("GPU:", torch.cuda.get_device_name(0)) |