Ver Fonte

add cuda lib path for nvidia container toolkit

Jeffrey Morgan há 1 ano atrás
pai
commit
1caa56128f
1 ficheiros alterados com 1 adições e 0 exclusões
  1. 1 0
      gpu/gpu_info_cuda.c

+ 1 - 0
gpu/gpu_info_cuda.c

@@ -9,6 +9,7 @@ const char *cuda_lib_paths[] = {
     "libnvidia-ml.so",
     "libnvidia-ml.so",
     "/usr/local/cuda/lib64/libnvidia-ml.so",
     "/usr/local/cuda/lib64/libnvidia-ml.so",
     "/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.1",
     "/usr/lib/wsl/lib/libnvidia-ml.so.1",  // TODO Maybe glob?
     "/usr/lib/wsl/lib/libnvidia-ml.so.1",  // TODO Maybe glob?
     NULL,
     NULL,
 };
 };