Add WSL2 path to nvidia-ml.so library

2023-12-15 20:16:02 -08:00 · 2023-12-15 20:16:02 -08:00 · 5646826a79
parent 3269535a4c
commit 5646826a79
1 changed files with 1 additions and 0 deletions
--- a/gpu/gpu_info_cuda.c
+++ b/gpu/gpu_info_cuda.c
@ -8,6 +8,7 @@
 const char *cuda_lib_paths[] = {
    "libnvidia-ml.so",
    "/usr/local/cuda/lib64/libnvidia-ml.so",
+    "/usr/lib/wsl/lib/libnvidia-ml.so.1",  // TODO Maybe glob?
    NULL,
 };
 #else