From e77c71b3f807a4870a3c8e0584b8b2ace2f7e515 Mon Sep 17 00:00:00 2001
From: Pablo Castellazzi <pablo.castellazzi@icloud.com>
Date: Fri, 2 Jan 2026 05:56:23 -0300
Subject: [PATCH] use default nvidia-smi as fallback when on WSL2

---
 scripts/install.sh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/scripts/install.sh b/scripts/install.sh
index 9c232400f..ddc0b0a47 100644
--- a/scripts/install.sh
+++ b/scripts/install.sh
@@ -173,6 +173,11 @@ fi
 # WSL2 only supports GPUs via nvidia passthrough
 # so check for nvidia-smi to determine if GPU is available
 if [ "$IS_WSL2" = true ]; then
+    # WSL2 has nvidia-smi installed by default,
+    # we add its folder to the end of the PATH as a fallback.
+    if [ -x /usr/lib/wsl/lib/nvidia-smi ]; then
+        export PATH=${PATH}:/usr/lib/wsl/lib
+    fi
     if available nvidia-smi && [ -n "$(nvidia-smi | grep -o "CUDA Version: [0-9]*\.[0-9]*")" ]; then
         status "Nvidia GPU detected."
     fi
@@ -239,7 +244,7 @@ CUDA_REPO_ERR_MSG="NVIDIA GPU detected, but your OS and Architecture are not sup
 # ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#fedora
 install_cuda_driver_yum() {
     status 'Installing NVIDIA repository...'
-    
+
     case $PACKAGE_MANAGER in
         yum)
             $SUDO $PACKAGE_MANAGER -y install yum-utils