[ROCM] Upgrade torch to 2.6 (#15244)

Signed-off-by: Sage Moore <sage@neuralmagic.com>

[ROCM] Upgrade torch to 2.6 (#15244)
Signed-off-by: Sage Moore <sage@neuralmagic.com>
2e0b4cfd · Sage Moore · GitHub · 10f55fe6 · 2e0b4cfd
Unverified Commit 2e0b4cfd authored Mar 20, 2025 by Sage Moore Committed by GitHub Mar 20, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

requirements/rocm-build.txt requirements/rocm-build.txt +4 -4

No files found.
--- a/requirements/rocm-build.txt
+++ b/requirements/rocm-build.txt
 # Common dependencies
 -r common.txt

--extra-index-url https://download.pytorch.org/whl/rocm6.2
-torch==2.5.1
-torchvision==0.20.1
-torchaudio==2.5.1
+--extra-index-url https://download.pytorch.org/whl/rocm6.2.4
+torch==2.6.0
+torchvision==0.21.0
+torchaudio==2.6.0

 cmake>=3.26
 packaging