From 60e78e2057e2197bb2b13d162c7cab26b0374dcb Mon Sep 17 00:00:00 2001
From: Gus Power <gus@infinitesidequests.com>
Date: Sun, 25 May 2025 09:15:54 +0100
Subject: a few snippets

---
 0002_rocm_check_gpu.py | 6 ++++++
 1 file changed, 6 insertions(+)
 create mode 100644 0002_rocm_check_gpu.py

(limited to '0002_rocm_check_gpu.py')

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
+
-- 
cgit v1.2.3