diff --git a/install.ps1 b/install.ps1 index 4dfd023af5..ce0211027e 100644 --- a/install.ps1 +++ b/install.ps1 @@ -1321,6 +1321,12 @@ shell.Run cmd, 0, False $ROCmTorchVisionUrl = "$amdRelBase/torchvision-0.24.1+rocm7.2.1-cp312-cp312-win_amd64.whl" $ROCmTorchAudioUrl = "$amdRelBase/torchaudio-2.9.1+rocm7.2.1-cp312-cp312-win_amd64.whl" $TorchIndexUrl = $null + } elseif ($ROCmVersion -and $ROCmVersion -match '^7\.1') { + $amdRelBase = "$amdWheelBase/rocm-rel-7.1.1" + $ROCmTorchWheelUrl = "$amdRelBase/torch-2.9.0+rocmsdk20251116-cp312-cp312-win_amd64.whl" + $ROCmTorchVisionUrl = "$amdRelBase/torchvision-0.24.0+rocmsdk20251116-cp312-cp312-win_amd64.whl" + $ROCmTorchAudioUrl = "$amdRelBase/torchaudio-2.9.0+rocmsdk20251116-cp312-cp312-win_amd64.whl" + $TorchIndexUrl = $null } if ($ROCmTorchWheelUrl) { substep "AMD ROCm $ROCmVersion (Python 3.12) -- AMD Windows torch wheel selected" "Cyan" diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py index 78885bfeef..a3e84d5885 100644 --- a/studio/install_python_stack.py +++ b/studio/install_python_stack.py @@ -71,6 +71,12 @@ _ROCM_WINDOWS_RELEASES: dict[tuple[int, int], tuple[str, str, str, str]] = { "0.24.1+rocm7.2.1", "2.9.1+rocm7.2.1", ), + (7, 1): ( + "rocm-rel-7.1.1", + "2.9.0+rocmsdk20251116", + "0.24.0+rocmsdk20251116", + "2.9.0+rocmsdk20251116", + ), } # bitsandbytes continuous-release_main wheels with the ROCm 4-bit GEMV fix