Skip to content

Commit 2e0b4cf

Browse files
authored
[ROCM] Upgrade torch to 2.6 (#15244)
Signed-off-by: Sage Moore <sage@neuralmagic.com>
1 parent 10f55fe commit 2e0b4cf

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

requirements/rocm-build.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
# Common dependencies
22
-r common.txt
33

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

99
cmake>=3.26
1010
packaging

0 commit comments

Comments
 (0)