From 1caa56128f644a55e174b207a22b54ce7a463233 Mon Sep 17 00:00:00 2001
From: Jeffrey Morgan <jmorganca@gmail.com>
Date: Fri, 5 Jan 2024 21:10:34 -0500
Subject: [PATCH] add cuda lib path for nvidia container toolkit

---
 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 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,
 };