0002_rocm_check_gpu.py


1
2
3
4
5
6

import torch
print(torch.cuda.is_available())  # Should be False
print(torch.backends.mps.is_available())  # Also False for AMD
print(torch.version.hip)  # Should show ROCm version
print(torch.version.cuda)  # Should be None