Move libnvidia-ml.so to CUDA_DSO_PATTERNS

numtide · Oct 12, 2024 · 958f5c0 · 958f5c0
1 parent 38b7dba
commit 958f5c0
Showing 1 changed file with 5 additions 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,11 @@ 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.*$"]