fuck nvidia driversHEADmain

1 files changed, 13 insertions, 0 deletions

diff --git a/.tmp_gpu_check.py b/.tmp_gpu_check.py
new file mode 100644
index 0000000..706491c
--- /dev/null
+++ b/.tmp_gpu_check.py
@@ -0,0 +1,13 @@
+import os
+print("ENV CUDA_VISIBLE_DEVICES:", os.environ.get("CUDA_VISIBLE_DEVICES"))
+try:
+    import torch
+    print("torch version:", torch.__version__)
+    print("torch.version.cuda:", getattr(torch.version, "cuda", None))
+    print("torch.cuda.is_available():", torch.cuda.is_available())
+    print("torch.cuda.device_count():", torch.cuda.device_count())
+    if torch.cuda.is_available():
+        for i in range(torch.cuda.device_count()):
+            print(f"[{i}]", torch.cuda.get_device_name(i))
+except Exception as e:
+    print("torch error:", repr(e))