[ROCm] Update the torch version in rocm_build.txt to use the official 2.10 release (#34387)
Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
@@ -1,7 +1,7 @@
|
|||||||
# 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
|
|||||||
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
|
||||||
|
|||||||
Reference in New Issue
Block a user