From 0429c5487727d06e42496f22bc5069b83217b3db Mon Sep 17 00:00:00 2001 From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com> Date: Mon, 12 May 2025 18:48:31 +0100 Subject: [PATCH] WIP --- .github/build_windows_packages.ps1 | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/.github/build_windows_packages.ps1 b/.github/build_windows_packages.ps1 index fd99137a..d366c7b7 100644 --- a/.github/build_windows_packages.ps1 +++ b/.github/build_windows_packages.ps1 @@ -68,13 +68,19 @@ Remove-Item $ffZip Remove-Item $ffDir.FullName -Recurse -Force Write-Host "[INFO] Installing PyTorch..." -$torchCmd = switch ($cuda) { - "cu124" { "pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124" } - "cu128" { "pip install torch==2.7.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128" } - default { Write-Error "Unsupported CUDA version: $cuda"; exit 1 } -} & ".\runtime\python.exe" -m ensurepip -& ".\runtime\python.exe" -c "$torchCmd" +switch ($cuda) { + "cu124" { + & ".\runtime\python.exe" -m pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124 + } + "cu128" { + & ".\runtime\python.exe" -m pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu128 + } + default { + Write-Error "Unsupported CUDA version: $cuda" + exit 1 + } +} Write-Host "[INFO] Installing dependencies..." & ".\runtime\python.exe" -m pip install -r extra-req.txt --no-deps --no-warn-script-location