Skip to content

GitLab

Unverified Commit 2e0b4cfd authored by Sage Moore's avatar Sage Moore Committed by GitHub
Browse files

[ROCM] Upgrade torch to 2.6 (#15244) 


Signed-off-by: default avatarSage Moore <sage@neuralmagic.com>
parent 10f55fe6
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
requirements/rocm-build.txt
View file @ 2e0b4cfd
# Common dependencies # Common dependencies
-r common.txt -r common.txt
--extra-index-url https://download.pytorch.org/whl/rocm6.2 --extra-index-url https://download.pytorch.org/whl/rocm6.2.4
torch==2.5.1 torch==2.6.0
torchvision==0.20.1 torchvision==0.21.0
torchaudio==2.5.1 torchaudio==2.6.0
cmake>=3.26 cmake>=3.26
packaging packaging
......
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