From 552db98bf111ade526b351d7d1a0f8f1f950d359 Mon Sep 17 00:00:00 2001 From: Daniel Hiltgen Date: Fri, 19 Jan 2024 13:23:29 -0800 Subject: [PATCH] More WSL paths --- gpu/gpu.go | 1 + 1 file changed, 1 insertion(+) diff --git a/gpu/gpu.go b/gpu/gpu.go index 451b7557..67bd8352 100644 --- a/gpu/gpu.go +++ b/gpu/gpu.go @@ -38,6 +38,7 @@ var CudaLinuxGlobs = []string{ "/usr/lib/x86_64-linux-gnu/nvidia/current/libnvidia-ml.so*", "/usr/lib/x86_64-linux-gnu/libnvidia-ml.so*", "/usr/lib/wsl/lib/libnvidia-ml.so*", + "/usr/lib/wsl/drivers/*/libnvidia-ml.so*", "/opt/cuda/lib64/libnvidia-ml.so*", "/opt/cuda/targets/x86_64-linux/lib/stubs/libnvidia-ml.so*", "/usr/lib*/libnvidia-ml.so*",