瀏覽代碼

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

Enhance GPU detection logic to support WSL2 environments
Timothy Jaeryang Baek 1 年之前
父節點
當前提交
ddd1a437ae
共有 1 個文件被更改,包括 2 次插入2 次删除
  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