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_check.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 .tmp_gpu_check.py (limited to '.tmp_gpu_check.py') 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)) -- cgit v1.2.3