From 7e2943597ddbadffd0e394e382a054e0ef66fed3 Mon Sep 17 00:00:00 2001 From: LeoBorcherding Date: Wed, 6 May 2026 17:55:23 -0500 Subject: [PATCH] feat: add ROCm 7.1.1 Windows wheel mapping AMD uses a different version string for 7.1.1 wheels: 2.9.0+rocmsdk20251116 (date-tagged) instead of +rocm7.1.1. Adds the 7.1.1 release folder to both install.ps1 and install_python_stack.py so users with ROCm 7.1 get ROCm torch instead of falling back to CPU. --- install.ps1 | 6 ++++++ studio/install_python_stack.py | 6 ++++++ 2 files changed, 12 insertions(+) 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