From 449c05a0ca3c0f493b66a4d061df8c6090619c6a Mon Sep 17 00:00:00 2001 From: John Calderon <81483067+johncalesp@users.noreply.github.com> Date: Tue, 13 Oct 2026 22:10:00 -0400 Subject: [PATCH] [CI][Tests] Upgrade torch version for Rubin prerelease (#61468) Signed-off-by: John Calderon Co-authored-by: Artem Perevedentsev --- requirements/rubin-prerelease.txt | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/requirements/rubin-prerelease.txt b/requirements/rubin-prerelease.txt index 6b416b48f6..720c088431 100644 --- a/requirements/rubin-prerelease.txt +++ b/requirements/rubin-prerelease.txt @@ -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