iHore/vLLM
8
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[CI][Tests] Upgrade torch version for Rubin prerelease (#61468)

Browse Source 
Signed-off-by: John Calderon <jcalderon@nvidia.com>
Co-authored-by: Artem Perevedentsev <aperevedents@nvidia.com>
This commit is contained in:
John Calderon
2026-10-14 10:10:00 +08:00
committed by GitHub
co-authored by Artem Perevedentsev
parent d5e62627a3
commit 449c05a0ca
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
requirements/rubin-prerelease.txt
+4 -5
View File
@@ -4,11 +4,10 @@
# (vllm-project/vllm#49735).
--extra-index-url https://download.pytorch.org/whl/nightly/cu134
# Match each architecture's torch pin to the latest torchvision wheel.
torch==2.15.0.dev20260928+cu134; platform_machine == "x86_64"
torch==2.15.0.dev20260926+cu134; platform_machine == "aarch64"
torchvision==0.30.0.dev20260928+cu134
torchaudio==2.11.0.dev20260928+cu134
# Use the same nightly pins for x86_64 and aarch64.
torch==2.16.0.dev20261009+cu134
torchvision==0.30.0.dev20261009+cu134
torchaudio==2.11.0.dev20261009+cu134
cuda-python==13.4.1
cuda-bindings==13.4.1