Browse Source

Merge pull request #575 from Mulugruntz/sgiffard/nvidia-gpu-on-wsl2

Enhance GPU detection logic to support WSL2 environments
Timothy Jaeryang Baek 1 year ago
parent
commit
ddd1a437ae
1 changed files with 2 additions and 2 deletions
  1. 2 2
      run-compose.sh

+ 2 - 2
run-compose.sh

@@ -11,8 +11,8 @@ TICK='\u2713'
 
 # Detect GPU driver
 get_gpu_driver() {
-    # Detect NVIDIA GPUs
-    if lspci | grep -i nvidia >/dev/null; then
+    # Detect NVIDIA GPUs using lspci or nvidia-smi
+    if lspci | grep -i nvidia >/dev/null || nvidia-smi >/dev/null 2>&1; then
         echo "nvidia"
         return
     fi