From 5646826a792ce07bbf4989210841d26b86f0de44 Mon Sep 17 00:00:00 2001
From: Daniel Hiltgen <daniel@ollama.com>
Date: Fri, 15 Dec 2023 20:16:02 -0800
Subject: [PATCH] Add WSL2 path to nvidia-ml.so library

---
 gpu/gpu_info_cuda.c | 1 +
 1 file changed, 1 insertion(+)

diff --git a/gpu/gpu_info_cuda.c b/gpu/gpu_info_cuda.c
index 710daa89..30aba037 100644
--- 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