Skip to content

Commit 4e19418

Browse files
committed
fix skipping the check for nvidia-smi
1 parent 872c68e commit 4e19418

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cuda_core/build_hooks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def _get_proper_cuda_bindings_major_version() -> str:
4545
m = re.search(r"CUDA Version:\s*([\d\.]+)", out.stdout.decode())
4646
if m:
4747
return m.group(1).split(".")[0]
48-
except FileNotFoundError:
48+
except (FileNotFoundError, subprocess.CalledProcessError):
4949
# the build machine has no driver installed
5050
pass
5151

0 commit comments

Comments
 (0)