Skip to content

GitLab

Unverified Commit 9e2cabdf authored by Sage Moore's avatar Sage Moore Committed by GitHub
Browse files

[ROCm] Update the torch version in rocm_build.txt to use the official 2.10 release (#34387) 


Signed-off-by: default avatarSage Moore <sage@neuralmagic.com>
parent ec8ab9d2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
requirements/rocm-build.txt
View file @ 9e2cabdf
# Common dependencies # Common dependencies
-r common.txt -r common.txt
--extra-index-url https://download.pytorch.org/whl/test/rocm7.0 --extra-index-url https://download.pytorch.org/whl/rocm7.1
torch==2.10.0 torch==2.10.0
torchvision==0.25.0 torchvision==0.25.0
torchaudio==2.10.0 torchaudio==2.10.0
...@@ -12,5 +12,5 @@ setuptools>=77.0.3,<80.0.0 ...@@ -12,5 +12,5 @@ setuptools>=77.0.3,<80.0.0
setuptools-scm>=8 setuptools-scm>=8
wheel wheel
jinja2>=3.1.6 jinja2>=3.1.6
amdsmi==6.4.3 amdsmi==7.0.2
timm>=1.0.17 timm>=1.0.17
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment