John CalderonandArtem Perevedentsev
449c05a0ca
Signed-off-by: John Calderon <jcalderon@nvidia.com> Co-authored-by: Artem Perevedentsev <aperevedents@nvidia.com>
15 lines
529 B
Plaintext
15 lines
529 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
|
|
|
|
# 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
|