Commit 5646826a authored by Daniel Hiltgen's avatar Daniel Hiltgen
Browse files

Add WSL2 path to nvidia-ml.so library

parent 3269535a
...@@ -8,6 +8,7 @@ ...@@ -8,6 +8,7 @@
const char *cuda_lib_paths[] = { 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/wsl/lib/libnvidia-ml.so.1", // TODO Maybe glob?
NULL, NULL,
}; };
#else #else
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment