From 9bdd1091a8045d74823d1f5de94b041805e42c3f Mon Sep 17 00:00:00 2001 From: Yuren Hao Date: Wed, 10 Sep 2025 14:58:32 -0500 Subject: fuck nvidia drivers --- .tmp_gpu_check2.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) create mode 100644 .tmp_gpu_check2.py (limited to '.tmp_gpu_check2.py') diff --git a/.tmp_gpu_check2.py b/.tmp_gpu_check2.py new file mode 100644 index 0000000..7555155 --- /dev/null +++ b/.tmp_gpu_check2.py @@ -0,0 +1,16 @@ +import os +print("ENV CUDA_VISIBLE_DEVICES:", os.environ.get("CUDA_VISIBLE_DEVICES")) +try: + import torch + print("torch:", torch.__version__, "cuda:", getattr(torch.version, "cuda", None)) + print("built_with_cuda:", torch.backends.cuda.is_built()) + print("device_count:", torch.cuda.device_count()) + print("is_available:", torch.cuda.is_available()) + if torch.cuda.device_count() > 0: + for i in range(torch.cuda.device_count()): + try: + print(f"[{i}]", torch.cuda.get_device_name(i)) + except Exception as e: + print(f"[{i}] name error:", e) +except Exception as e: + print("torch error:", repr(e)) -- cgit v1.2.3