add cuda lib path for nvidia container toolkit

1caa5612 · Jeffrey Morgan · 0101e76d · 1caa5612
Commit 1caa5612 authored Jan 05, 2024 by Jeffrey Morgan
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

gpu/gpu_info_cuda.c gpu/gpu_info_cuda.c +1 -0

No files found.
--- a/gpu/gpu_info_cuda.c
+++ b/gpu/gpu_info_cuda.c
@@ -9,6 +9,7 @@ const char *cuda_lib_paths[] = {
    "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/libnvidia-ml.so.1",
    "/usr/lib/wsl/lib/libnvidia-ml.so.1",  // TODO Maybe glob?
    NULL,
 };