Update ImageStudioAlternativo.sh

ImageStudioAlternativo.sh

@@ -287,6 +287,20 @@ fi
 
 log_success "ComfyUI instalado"
 
+# 4. Corrigir PyTorch para RTX 5090 se detectado
+log_info "[4/6] Verificando GPU..."
+
+GPU_INFO=$(nvidia-smi --query-gpu=name --format=csv,noheader 2>/dev/null || echo "")
+
+if [[ "$GPU_INFO" == *"5090"* ]] || [[ "$GPU_INFO" == *"5080"* ]]; then
+    log_warn "RTX 5090/5080 detectada - instalando PyTorch novo"
+    python -m pip install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
+else
+    log_info "GPU: ${GPU_INFO:-Não detectada}"
+fi
+
+log_success "PyTorch configurado"
+
 # 5. Baixar modelos
 log_info "[5/6] Baixando modelos..."