# 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