Enhance GPU detection logic to support WSL2 environments
@@ -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