Ethanfel
d8b3972bdc
pip install -r requirements.txt can pull CPU-only torchvision via transitive dependencies (timm, ultralytics). Adding --extra-index-url with the CUDA wheel index ensures all torch packages stay on the correct build. Applied to both Linux and Windows setup scripts. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
92 lines
4.7 KiB
PowerShell
92 lines
4.7 KiB
PowerShell
# 8-cut Windows setup script
|
|
# Run once: powershell -ExecutionPolicy Bypass -File setup-windows.ps1
|
|
#
|
|
# Prerequisites: Python 3.11+ must be installed and on PATH
|
|
# https://www.python.org/downloads/
|
|
|
|
$ErrorActionPreference = "Stop"
|
|
trap { Write-Host "`n$_" -ForegroundColor Red; Read-Host "Press Enter to close"; exit 1 }
|
|
$root = Split-Path -Parent $MyInvocation.MyCommand.Path
|
|
|
|
Write-Host "=== 8-cut Windows Setup ===" -ForegroundColor Cyan
|
|
|
|
# ── Virtual environment ───────────────────────────────────
|
|
$venvDir = Join-Path $root ".venv"
|
|
if (Test-Path (Join-Path $venvDir "Scripts\python.exe")) {
|
|
Write-Host "`nVirtual environment already exists, activating..." -ForegroundColor Green
|
|
} else {
|
|
Write-Host "`nCreating virtual environment..."
|
|
python -m venv $venvDir
|
|
Write-Host "Virtual environment created at $venvDir" -ForegroundColor Green
|
|
}
|
|
& "$venvDir\Scripts\Activate.ps1"
|
|
|
|
# ── PyTorch ───────────────────────────────────────────────
|
|
# Detect NVIDIA GPU via nvidia-smi
|
|
$hasNvidia = Get-Command nvidia-smi -ErrorAction SilentlyContinue
|
|
if ($hasNvidia) {
|
|
$torchIndex = "https://download.pytorch.org/whl/cu128"
|
|
Write-Host "`nNVIDIA GPU detected — using CUDA 12.8 PyTorch index" -ForegroundColor Green
|
|
} else {
|
|
$torchIndex = "https://download.pytorch.org/whl/cpu"
|
|
Write-Host "`nNo NVIDIA GPU detected — using CPU-only PyTorch index" -ForegroundColor Yellow
|
|
}
|
|
# 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 ───────────────────────────────────────────
|
|
Write-Host "`nInstalling project dependencies..."
|
|
pip install -r (Join-Path $root "requirements.txt") --extra-index-url $torchIndex
|
|
|
|
# ── libmpv ────────────────────────────────────────────────
|
|
$mpvDll = Join-Path $root "libmpv-2.dll"
|
|
if (Test-Path $mpvDll) {
|
|
Write-Host "`nlibmpv-2.dll already present, skipping." -ForegroundColor Green
|
|
} else {
|
|
Write-Host "`nDownloading libmpv..."
|
|
$release = Invoke-RestMethod "https://api.github.com/repos/shinchiro/mpv-winbuild-cmake/releases/latest"
|
|
$asset = $release.assets | Where-Object { $_.name -like "mpv-dev-x86_64-v3-*" } | Select-Object -First 1
|
|
$tmpFile = Join-Path $root "mpv-dev.7z"
|
|
Invoke-WebRequest $asset.browser_download_url -OutFile $tmpFile
|
|
7z x $tmpFile -o"$root\mpv-dev" -y | Out-Null
|
|
Copy-Item "$root\mpv-dev\libmpv-2.dll" $root
|
|
Remove-Item $tmpFile -Force
|
|
Remove-Item "$root\mpv-dev" -Recurse -Force
|
|
Write-Host "libmpv-2.dll downloaded." -ForegroundColor Green
|
|
}
|
|
|
|
# ── ffmpeg ────────────────────────────────────────────────
|
|
$ffmpeg = Join-Path $root "ffmpeg.exe"
|
|
if (Test-Path $ffmpeg) {
|
|
Write-Host "`nffmpeg.exe already present, skipping." -ForegroundColor Green
|
|
} else {
|
|
$onPath = Get-Command ffmpeg -ErrorAction SilentlyContinue
|
|
if ($onPath) {
|
|
Write-Host "`nffmpeg found on PATH: $($onPath.Source)" -ForegroundColor Green
|
|
} else {
|
|
Write-Host "`nDownloading ffmpeg..."
|
|
$ffUrl = "https://github.com/BtbN/FFmpeg-Builds/releases/download/latest/ffmpeg-master-latest-win64-gpl.zip"
|
|
$tmpZip = Join-Path $root "ffmpeg.zip"
|
|
Invoke-WebRequest $ffUrl -OutFile $tmpZip
|
|
Expand-Archive $tmpZip -DestinationPath "$root\ffmpeg-tmp" -Force
|
|
$bin = Get-ChildItem -Path "$root\ffmpeg-tmp" -Recurse -Filter ffmpeg.exe | Select-Object -First 1
|
|
Copy-Item "$($bin.DirectoryName)\ffmpeg.exe" $root
|
|
Copy-Item "$($bin.DirectoryName)\ffprobe.exe" $root
|
|
Remove-Item $tmpZip -Force
|
|
Remove-Item "$root\ffmpeg-tmp" -Recurse -Force
|
|
Write-Host "ffmpeg.exe downloaded." -ForegroundColor Green
|
|
}
|
|
}
|
|
|
|
# ── Verify ────────────────────────────────────────────────
|
|
Write-Host "`n--- Verification ---" -ForegroundColor Cyan
|
|
python -c "import torch; print('PyTorch', torch.__version__, 'CUDA', torch.version.cuda)"
|
|
python -c "import sklearn, librosa, torchaudio; print('All imports OK')"
|
|
|
|
Write-Host "`n=== Setup complete ===" -ForegroundColor Cyan
|
|
Write-Host "Run 8-cut with: .venv\Scripts\python.exe main.py"
|
|
Write-Host "Or double-click: 8cut.bat"
|
|
Read-Host "`nPress Enter to close"
|