瀏覽代碼

add more search paths for cuda libs

Jeffrey Morgan 1 年之前
父節點
當前提交
be721ca0df
共有 1 個文件被更改,包括 8 次插入1 次删除
  1. 8 1
      gpu/gpu_info_cuda.c

+ 8 - 1
gpu/gpu_info_cuda.c

@@ -7,10 +7,17 @@
 #ifndef _WIN32
 const char *cuda_lib_paths[] = {
     "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/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/libnvidia-ml.so",
     "/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,
 };
 #else