8e443871f0
Signed-off-by: Shang Wang <samshang.wang@mail.utoronto.ca> Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com> Co-authored-by: Tyler Michael Smith <tlrmchlsmth@gmail.com> Co-authored-by: Shengqi Chen <harry-chen@outlook.com>
16 lines
641 B
Plaintext
16 lines
641 B
Plaintext
# Public CUDA 13.4 prerelease PyTorch nightly pins for Rubin bring-up.
|
|
# Installed when INSTALL_RUBIN_PRERELEASE=true (see docker/Dockerfile).
|
|
# TODO(rubin-bringup): delete once Rubin APIs ship in regular PyTorch wheels
|
|
# (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
|
|
|
|
cuda-python==13.4.1
|
|
cuda-bindings==13.4.1
|
|
nixl==1.4.1
|