Browse Source

add more search paths for cuda libs

Jeffrey Morgan 1 year ago
parent
commit
be721ca0df
1 changed files with 8 additions and 1 deletions
  1. 8 1
      gpu/gpu_info_cuda.c

+ 8 - 1
gpu/gpu_info_cuda.c

@@ -7,10 +7,17 @@
 #ifndef _WIN32
 #ifndef _WIN32
 const char *cuda_lib_paths[] = {
 const char *cuda_lib_paths[] = {
     "libnvidia-ml.so",
     "libnvidia-ml.so",
+    "/usr/lib/wsl/lib/libnvidia-ml.so",  // TODO Maybe glob?
+    "/usr/lib/wsl/lib/libnvidia-ml.so.1",
     "/usr/local/cuda/lib64/libnvidia-ml.so",
     "/usr/local/cuda/lib64/libnvidia-ml.so",
+    "/usr/lib/libnvidia-ml.so",
+    "/usr/lib/libnvidia-ml.so.1",
     "/usr/lib/x86_64-linux-gnu/nvidia/current/libnvidia-ml.so",
     "/usr/lib/x86_64-linux-gnu/nvidia/current/libnvidia-ml.so",
+    "/usr/lib/x86_64-linux-gnu/libnvidia-ml.so",
     "/usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1",
     "/usr/lib/x86_64-linux-gnu/libnvidia-ml.so.1",
-    "/usr/lib/wsl/lib/libnvidia-ml.so.1",  // TODO Maybe glob?
+    "/usr/lib/aarch64-linux-gnu/nvidia/current/libnvidia-ml.so",
+    "/usr/lib/aarch64-linux-gnu/libnvidia-ml.so",
+    "/usr/lib/aarch64-linux-gnu/libnvidia-ml.so.1",
     NULL,
     NULL,
 };
 };
 #else
 #else