diff options
| author | Gus Power <gus@infinitesidequests.com> | 2025-05-25 09:15:54 +0100 |
|---|---|---|
| committer | Gus Power <gus@infinitesidequests.com> | 2025-05-25 09:15:54 +0100 |
| commit | 60e78e2057e2197bb2b13d162c7cab26b0374dcb (patch) | |
| tree | 908cf0558bd239b9d6a5d5c3f7ecb47243b56e0f /0002_rocm_check_gpu.py | |
a few snippets
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 + |