# 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

# 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
nixl==1.4.1