adbrasi
/

importantFiles

adbrasi commited on Nov 9, 2025

Commit

26bcd7a

verified ·

1 Parent(s): 3283bab

Update ImageStudioArrakis2.sh

Files changed (1) hide show

ImageStudioArrakis2.sh CHANGED Viewed

@@ -298,7 +298,7 @@ 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 ${PIP_FLAGS} --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128
 else
     log_info "GPU: ${GPU_INFO:-Não detectada}"
 fi

 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