From 4e194188640a4eea83db581b616b54a26d3709aa Mon Sep 17 00:00:00 2001 From: Leo Fang Date: Fri, 3 Oct 2025 20:15:41 +0000 Subject: [PATCH] fix skipping the check for nvidia-smi --- cuda_core/build_hooks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cuda_core/build_hooks.py b/cuda_core/build_hooks.py index 7c5fd4672..6a55b6edd 100644 --- a/cuda_core/build_hooks.py +++ b/cuda_core/build_hooks.py @@ -45,7 +45,7 @@ def _get_proper_cuda_bindings_major_version() -> str: m = re.search(r"CUDA Version:\s*([\d\.]+)", out.stdout.decode()) if m: return m.group(1).split(".")[0] - except FileNotFoundError: + except (FileNotFoundError, subprocess.CalledProcessError): # the build machine has no driver installed pass