pyrox.dev
1diff --git a/third_party/nvidia/backend/driver.py b/third_party/nvidia/backend/driver.py
2index d088ec092..625de2db8 100644
3--- a/third_party/nvidia/backend/driver.py
4+++ b/third_party/nvidia/backend/driver.py
5@@ -23,6 +23,9 @@ def libcuda_dirs():
6 if env_libcuda_path:
7 return [env_libcuda_path]
8
9+ if os.path.exists("@libcudaStubsDir@"):
10+ return ["@libcudaStubsDir@"]
11+
12 libs = subprocess.check_output(["/sbin/ldconfig", "-p"]).decode()
13 # each line looks like the following:
14 # libcuda.so.1 (libc6,x86-64) => /lib/x86_64-linux-gnu/libcuda.so.1