fix skipping the check for nvidia-smi (NVIDIA#1084)

leofang · web-flow · commit b09d7ed1f92e · 2025-10-03T20:00:27.000-04:00
diff --git 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
 
