Move libnvidia-ml.so to CUDA_DSO_PATTERNS

numtide · Oct 12, 2024 · 286684c · 286684c
1 parent 38b7dba
commit 286684c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/nixglhost.py b/src/nixglhost.py
@@ -191,7 +191,6 @@ def from_json(cls, j: str):
     "libnvidia-glsi\\.so.*$",
     "libnvidia-glvkspirv\\.so.*$",
     "libnvidia-gpucomp\\.so.*$",
-    "libnvidia-ml\\.so.*$",
     "libnvidia-ngx\\.so.*$",
     "libnvidia-nvvm\\.so.*$",
     "libnvidia-opencl\\.so.*$",
@@ -229,7 +228,7 @@ def from_json(cls, j: str):
     "libdxcore\\.so.*$",
 ]
 
-CUDA_DSO_PATTERNS = ["libcudadebugger\\.so.*$", "libcuda\\.so.*$"]
+CUDA_DSO_PATTERNS = ["libcudadebugger\\.so.*$", "libcuda\\.so.*$", "libnvidia-ml\\.so.*$"]
 
 GLX_DSO_PATTERNS = ["libGLX_nvidia\\.so.*$"]