xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-15 23:35:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ROCM] Upgrade torch to 2.6 (#15244)

Browse Source 
Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
Sage Moore 2025-03-20 19:17:33 -07:00 committed by GitHub
parent 10f55fe6c5
commit 2e0b4cfde0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
requirements/rocm-build.txt
View File

@ -1,10 +1,10 @@
# 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