File size: 916 Bytes
4fcca49 c61e4fd 4fcca49 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | $ErrorActionPreference = "Stop"
$Root = Split-Path -Parent $PSScriptRoot
Set-Location $Root
if (-not (Test-Path ".venv")) { py -3.12 -m venv .venv }
$Py = Join-Path $Root ".venv\Scripts\python.exe"
& $Py -m pip install --upgrade pip
# Keep ORT on CUDA 12.x. ORT >=1.27 defaults to CUDA 13.
& $Py -m pip install "onnxruntime-gpu[cuda,cudnn]>=1.24,<1.27" torch huggingface_hub numpy pyzmq
if (-not (Test-Path ".vendor\NitroGen")) {
New-Item -ItemType Directory -Force ".vendor" | Out-Null
git clone https://github.com/MineDojo/NitroGen.git .vendor/NitroGen
git -C .vendor/NitroGen checkout 32608444660950ffda95e1e57c79632ad65bea10
}
& $Py -m pip install -e .vendor/NitroGen
Write-Host "`nReady. Benchmark:" -ForegroundColor Green
Write-Host ".\.venv\Scripts\python.exe scripts\benchmark_dit.py"
Write-Host "Gameplay server (fast profile):"
Write-Host ".\.venv\Scripts\python.exe scripts\serve_onnx.py --steps 4"
|