diff options
Diffstat (limited to '0002_rocm_check_gpu.py')
| -rw-r--r-- | 0002_rocm_check_gpu.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/0002_rocm_check_gpu.py b/0002_rocm_check_gpu.py new file mode 100644 index 0000000..8c92adb --- /dev/null +++ b/0002_rocm_check_gpu.py @@ -0,0 +1,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 + |