diff --git a/gpu/gpu_info_cuda.c b/gpu/gpu_info_cuda.c index c81293aa..f071f909 100644 --- 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, };