Ethanfel/8-cut
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: always reinstall torch stack on Windows re-runs

Browse Source 
Remove the "skip if torch exists" guard so re-running the setup script
fixes a broken torchvision install.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-04-19 16:09:44 +02:00
parent a3c657c66e
commit 3c3b1d74bb
1 changed files with 11 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
setup-windows.ps1
+8 -10
View File
@@ -22,21 +22,19 @@ if (Test-Path (Join-Path $venvDir "Scripts\python.exe")) {
& "$venvDir\Scripts\Activate.ps1" & "$venvDir\Scripts\Activate.ps1"
# ── PyTorch ─────────────────────────────────────────────── # ── PyTorch ───────────────────────────────────────────────
$hasTorch = python -c "import torch" 2>&1
if ($LASTEXITCODE -eq 0) {
Write-Host "`nPyTorch already installed, skipping." -ForegroundColor Green
} else {
# Detect NVIDIA GPU via nvidia-smi # Detect NVIDIA GPU via nvidia-smi
$hasNvidia = Get-Command nvidia-smi -ErrorAction SilentlyContinue $hasNvidia = Get-Command nvidia-smi -ErrorAction SilentlyContinue
if ($hasNvidia) { if ($hasNvidia) {
Write-Host "`nNVIDIA GPU detected — installing PyTorch with CUDA 12.8..." -ForegroundColor Green $torchIndex = "https://download.pytorch.org/whl/cu128"
pip install torch torchaudio torchvision --index-url https://download.pytorch.org/whl/cu128 Write-Host "`nNVIDIA GPU detected — using CUDA 12.8 PyTorch index" -ForegroundColor Green
} else { } else {
Write-Host "`nNo NVIDIA GPU detected — installing CPU-only PyTorch..." -ForegroundColor Yellow $torchIndex = "https://download.pytorch.org/whl/cpu"
Write-Host "(Audio scanning will work but will be slower without GPU)" -ForegroundColor Yellow Write-Host "`nNo NVIDIA GPU detected — using CPU-only PyTorch index" -ForegroundColor Yellow
pip install torch torchaudio torchvision --index-url https://download.pytorch.org/whl/cpu
}
} }
# Always install/upgrade torch stack from correct index
# (pip install is a no-op if already at the right version)
Write-Host "Installing PyTorch + torchaudio + torchvision..."
pip install torch torchaudio torchvision --index-url $torchIndex
# ── Python deps ─────────────────────────────────────────── # ── Python deps ───────────────────────────────────────────
Write-Host "`nInstalling project dependencies..." Write-Host "`nInstalling project dependencies..."