Add additional guard for "no NVIDIA driver"

bitsandbytes-foundation · Mar 11, 2024 · 2416dd3 · 2416dd3
1 parent 62249b4
commit 2416dd3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -13,6 +13,11 @@ def pytest_runtest_call(item):
         if str(ae) == "Torch not compiled with CUDA enabled":
             pytest.skip("Torch not compiled with CUDA enabled")
         raise
+    except RuntimeError as re:
+        # CUDA-enabled Torch build, but no CUDA-capable device found
+        if "Found no NVIDIA driver on your system" in str(re):
+            pytest.skip("No NVIDIA driver found")
+        raise
 
 
 @pytest.fixture(scope="session")