Enhance GPU detection logic to support WSL2 environments

This commit updates the GPU detection logic in the setup script to accommodate environments where NVIDIA GPUs might not be directly listed by `lspci`, such as in Windows Subsystem for Linux 2 (WSL2). The revised logic now attempts to identify NVIDIA GPUs using both `lspci` and `nvidia-smi`, ensuring broader compatibility and recognizing GPUs that are accessible in WSL2. This change allows the script to correctly detect NVIDIA GPUs in a wider range of setups, improving user experience and script robustness in diverse environments.

run-compose.sh

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