Browse Source

Add WSL2 path to nvidia-ml.so library

Daniel Hiltgen 1 year ago
parent
commit
5646826a79
1 changed files with 1 additions and 0 deletions
  1. 1 0
      gpu/gpu_info_cuda.c

+ 1 - 0
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