From 0c85b98fe3ea707bdb770ffec2fb2d82d7803b26 Mon Sep 17 00:00:00 2001
From: Jiangzhou Li <605858621@qq.com>
Date: Tue, 18 Mar 2025 17:33:44 -0700
Subject: [PATCH] fix: nvidia-smi path issue for root user in WSL

---
 install.sh           | 4 ++++
 install.sh.sha256sum | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/install.sh b/install.sh
index b2a2d5e..186343f 100755
--- a/install.sh
+++ b/install.sh
@@ -172,6 +172,10 @@ detect_device() {
       warn "NVIDIA GPU detected but CUDA is not installed. Please install CUDA."
     fi
     DEVICE="cuda"
+    # Create a symlink for nvidia-smi to allow root users in WSL to detect GPU information.
+    if [ -f "/usr/lib/wsl/lib/nvidia-smi" ] && [ ! -e "/usr/local/bin/nvidia-smi" ]; then
+      $SUDO ln -s /usr/lib/wsl/lib/nvidia-smi /usr/local/bin/nvidia-smi
+    fi
   fi
 
   if check_command "mthreads-gmi"; then
diff --git a/install.sh.sha256sum b/install.sh.sha256sum
index b01f8ce..0751bb7 100644
--- a/install.sh.sha256sum
+++ b/install.sh.sha256sum
@@ -1 +1 @@
-aa6ebe9416c158f756e6efa0741f2d9bee337d4b2e6e6fde92852c6d624e94eb  install.sh
+692c0927451c841b7a4963e457788c37fc09a0a414889e61c3ea7d84cd8b90b4  install.sh