| 1 | # Public CUDA 13.4 prerelease PyTorch nightly pins for Rubin bring-up. |
| 2 | # Installed when INSTALL_RUBIN_PRERELEASE=true (see docker/Dockerfile). |
| 3 | # TODO(rubin-bringup): delete once Rubin APIs ship in regular PyTorch wheels |
| 4 | # (vllm-project/vllm#49735). |
| 5 | --extra-index-url https://download.pytorch.org/whl/nightly/cu134 |
| 6 | --extra-index-url https://flashinfer.ai/whl/cu134 |
| 7 | --extra-index-url https://flashinfer.ai/whl |
| 8 | |
| 9 | # Match each architecture's torch pin to the latest torchvision wheel. |
| 10 | torch==2.15.0.dev20260909+cu134; platform_machine == "x86_64" |
| 11 | torch==2.15.0.dev20260908+cu134; platform_machine == "aarch64" |
| 12 | torchvision==0.30.0.dev20260909+cu134 |
| 13 | torchaudio==2.11.0.dev20260911+cu134 |
| 14 | |
| 15 | # docker/Dockerfile strips these packages from requirements/cuda.txt whenever |
| 16 | # INSTALL_RUBIN_PRERELEASE=true, so they must be pinned here instead to |
| 17 | # still get installed. |
| 18 | nvidia-cutlass-dsl[cu13]==4.8.0.dev0 |
| 19 | flashinfer-python==0.7.0.post1 |
| 20 | quack-kernels==0.6.5 |
| 21 | |
| 22 | cuda-python==13.4.1 |
| 23 | cuda-bindings==13.4.1 |
| 24 | flashinfer-jit-cache==0.7.0.post1+cu134 |
| 25 | flashinfer-cubin==0.7.0.post1 |
| 26 | nixl==1.4.1 |