Fix check

mdboom · mdboom · commit eaa2e6282b68 · 2026-02-06T12:35:24.000-05:00
diff --git a/cuda_bindings/cuda/bindings/_test_helpers/arch_check.py b/cuda_bindings/cuda/bindings/_test_helpers/arch_check.py
@@ -19,7 +19,7 @@ def hardware_supports_nvml():
     nvml.init_v2()
     try:
         nvml.system_get_driver_branch()
-    except nvml.NotSupportedError:
+    except (nvml.NotSupportedError, nvml.UnknownError):
         return False
     else:
         return True
