Astral GPU indexes Pre-built, GPU-enabled wheels for the Python ecosystem.
Pre-built wheels for popular GPU-enabled packages in the PyTorch ecosystem, like FlashAttention , across a range of Python versions, CPU architectures, CUDA versions, and PyTorch versions, based on open source build pipelines.
Following the PyTorch convention, the Astral GPU indexes use a separate index URL for each supported CUDA version (e.g., /simple/cu128
), and publish a separate wheel version for each supported CUDA and PyTorch combination (e.g., +cu.12.8.torch.2.11
).
At present, the Astral GPU indexes include builds for the following packages:
Available indexes CPU cpu / 3 packages CUDA 11.8 cu118 / 1 package CUDA 12.1 cu121 / 12 packages CUDA 12.4 cu124 / 14 packages CUDA 12.6 cu126 / 18 packages CUDA 12.8 cu128 / 22 packages CUDA 12.9 cu129 / 18 packages CUDA 13.0 cu130 / 17 packages CUDA 13.2 cu132 / 8 packages
Using an index with uv
Use `uv add`
to pin a package to the Astral GPU index.
The uv pip interface
Add the Astral GPU index as an extra --index
at install time.
pip
Add the Astral GPU index as an extra --extra-index-url
at install time.
Manifest
Package versions available in the CUDA 12.6 index. adan 0.0.2+cu.12.6.torch.2.11 , 0.0.2+cu.12.6.torch.2.10 , 0.0.2+cu.12.6.torch.2.9 , 0.0.2+cu.12.6.torch.2.8 , 0.0.2+cu.12.6.torch.2.7 , 0.0.2+cu.12.6.torch.2.6 causal-conv1d 1.6.2.post1+cu.12.6.torch.2.11 , 1.6.2.post1+cu.12.6.torch.2.10 , 1.6.2.post1+cu.12.6.torch.2.9 , 1.6.2.post1+cu.12.6.torch.2.8 , 1.6.2.post1+cu.12.6.torch.2.7 , 1.6.2.post1+cu.12.6.torch.2.6 , 1.6.1+cu.12.6.torch.2.10 , 1.6.1+cu.12.6.torch.2.9 , 1.6.1+cu.12.6.torch.2.8 , 1.6.1+cu.12.6.torch.2.7 , 1.6.1+cu.12.6.torch.2.6 , 1.6.0+cu12.6torch2.10.0cxx11abitrue , 1.6.0+cu12.6torch2.9.0cxx11abitrue , 1.6.0+cu12.6torch2.8.0cxx11abitrue , 1.6.0+cu12.6torch2.7.1cxx11abitrue , 1.6.0+cu12.6torch2.6.0cxx11abitrue , 1.5.4+cu12.6torch2.9.0cxx11abitrue , 1.5.4+cu12.6torch2.8.0cxx11abitrue , 1.5.4+cu12.6torch2.7.1cxx11abitrue , 1.5.4+cu12.6torch2.6.0cxx11abitrue deepspeed 0.19.2+cu.12.6.torch.2.12 , 0.19.2+cu.12.6.torch.2.11 , 0.19.2+cu.12.6.torch.2.10 , 0.19.2+cu.12.6.torch.2.9 , 0.19.2+cu.12.6.torch.2.8 , 0.19.2+cu.12.6.torch.2.7 , 0.19.1+cu.12.6.torch.2.12 , 0.19.1+cu.12.6.torch.2.11 , 0.19.1+cu.12.6.torch.2.10 , 0.19.1+cu.12.6.torch.2.9 , 0.19.1+cu.12.6.torch.2.8 , 0.19.1+cu.12.6.torch.2.7 , 0.19.0+cu.12.6.torch.2.12 , 0.19.0+cu.12.6.torch.2.11 , 0.19.0+cu.12.6.torch.2.10 , 0.19.0+cu.12.6.torch.2.9 , 0.19.0+cu.12.6.torch.2.8 , 0.19.0+cu.12.6.torch.2.7 , 0.18.9+cu.12.6.torch.2.11 , 0.18.9+cu.12.6.torch.2.10 , 0.18.9+cu.12.6.torch.2.9 , 0.18.9+cu.12.6.torch.2.8 , 0.18.9+cu.12.6.torch.2.7 , 0.18.8+cu.12.6.torch.2.10 , 0.18.8+cu.12.6.torch.2.9 , 0.18.8+cu.12.6.torch.2.8 , 0.18.8+cu.12.6.torch.2.7 , 0.18.7+cu.12.6.torch.2.10 , 0.18.7+cu.12.6.torch.2.9 , 0.18.7+cu.12.6.torch.2.8 , 0.18.7+cu.12.6.torch.2.7 , 0.18.6+cu.12.6.torch.2.10 , 0.18.6+cu.12.6.torch.2.9 , 0.18.6+cu.12.6.torch.2.8 , 0.18.6+cu.12.6.torch.2.7 , 0.18.5+cu.12.6.torch.2.10 , 0.18.5+cu.12.6.torch.2.9 , 0.18.5+cu.12.6.torch.2.8 , 0.18.5+cu.12.6.torch.2.7 , 0.18.4+cu12.6torch2.10.0cxx11abitrue , 0.18.4+cu12.6torch2.9.0cxx11abitrue , 0.18.4+cu12.6torch2.8.0cxx11abitrue , 0.18.4+cu12.6torch2.7.1cxx11abitrue , 0.18.3+cu12.6torch2.9.0cxx11abitrue , 0.18.3+cu12.6torch2.8.0cxx11abitrue , 0.18.3+cu12.6torch2.7.1cxx11abitrue , 0.18.2+cu12.6torch2.9.0cxx11abitrue , 0.18.2+cu12.6torch2.8.0cxx11abitrue , 0.18.2+cu12.6torch2.7.1cxx11abitrue , 0.18.1+cu12.6torch2.9.0cxx11abitrue , 0.18.1+cu12.6torch2.8.0cxx11abitrue , 0.18.1+cu12.6torch2.7.1cxx11abitrue detectron2 0.6+cu.12.6.torch.2.12 , 0.6+cu.12.6.torch.2.11 , 0.6+cu.12.6.torch.2.10 , 0.6+cu.12.6.torch.2.9 , 0.6+cu.12.6.torch.2.8 , 0.6+cu.12.6.torch.2.7 , 0.6+cu.12.6.torch.2.6 flash-attn 2.8.3.post1+cu.12.6.torch.2.12 , 2.8.3.post1+cu.12.6.torch.2.11 , 2.8.3.post1+cu.12.6.torch.2.10 , 2.8.3.post1+cu.12.6.torch.2.9 , 2.8.3.post1+cu.12.6.torch.2.8 , 2.8.3.post1+cu.12.6.torch.2.7 , 2.8.3.post1+cu.12.6.torch.2.6 , 2.8.3+cu.12.6.torch.2.11 , 2.8.3+cu.12.6.torch.2.10 , 2.8.3+cu.12.6.torch.2.9 , 2.8.3+cu.12.6.torch.2.8 , 2.8.3+cu.12.6.torch.2.7 , 2.8.3+cu.12.6.torch.2.6 mamba-ssm 2.3.2.post1+cu.12.6.torch.2.11 , 2.3.2.post1+cu.12.6.torch.2.10 , 2.3.2.post1+cu.12.6.torch.2.9 , 2.3.2.post1+cu.12.6.torch.2.8 , 2.3.2.post1+cu.12.6.torch.2.7 , 2.3.2.post1+cu.12.6.torch.2.6 , 2.3.1+cu.12.6.torch.2.10 , 2.3.1+cu.12.6.torch.2.9 , 2.3.1+cu.12.6.torch.2.8 , 2.3.1+cu.12.6.torch.2.7 , 2.3.1+cu.12.6.torch.2.6 , 2.3.0+cu12.6torch2.10.0cxx11abitrue , 2.3.0+cu12.6torch2.9.1cxx11abitrue , 2.3.0+cu12.6torch2.8.0cxx11abitrue , 2.3.0+cu12.6torch2.7.1cxx11abitrue , 2.3.0+cu12.6torch2.6.0cxx11abitrue , 2.2.6.post3+cu12.6torch2.9.0cxx11abitrue , 2.2.6.post3+cu12.6torch2.8.0cxx11abitrue , 2.2.6.post3+cu12.6torch2.7.1cxx11abitrue , 2.2.6.post3+cu12.6torch2.6.0cxx11abitrue mmcv 2.2.0+cu.12.6.torch.2.11 , 2.2.0+cu.12.6.torch.2.10 , 2.2.0+cu.12.6.torch.2.9 , 2.2.0+cu.12.6.torch.2.8 , 2.2.0+cu.12.6.torch.2.7 , 2.2.0+cu.12.6.torch.2.6 pycuda 2026.1+cu.12.6 , 2025.1.3+cu.12.6 , 2025.1.2+cu12.6 pyg-lib 0.8.0+cu.12.6.torch.2.13 , 0.8.0+cu.12.6.torch.2.12 , 0.8.0+cu.12.6.torch.2.11 , 0.8.0+cu.12.6.torch.2.10 pytorch3d 0.7.9+cu.12.6.torch.2.12 , 0.7.9+cu.12.6.torch.2.11 , 0.7.9+cu.12.6.torch.2.10 , 0.7.9+cu.12.6.torch.2.9 , 0.7.9+cu.12.6.torch.2.8 , 0.7.9+cu.12.6.torch.2.7 , 0.7.9+cu.12.6.torch.2.6 , 0.7.8+cu12.6torch2.9.0cxx11abitrue , 0.7.8+cu12.6torch2.8.0cxx11abitrue , 0.7.8+cu12.6torch2.7.1cxx11abitrue , 0.7.8+cu12.6torch2.6.0cxx11abitrue sageattention 2.2.0+cu.12.6.torch.2.11 , 2.2.0+cu.12.6.torch.2.10 , 2.2.0+cu.12.6.torch.2.9 , 2.2.0+cu.12.6.torch.2.8 , 2.2.0+cu.12.6.torch.2.7 , 2.2.0+cu.12.6.torch.2.6 torch-scatter 2.1.2+cu.12.6.torch.2.11 , 2.1.2+cu.12.6.torch.2.10 , 2.1.2+cu.12.6.torch.2.9 , 2.1.2+cu.12.6.torch.2.8 , 2.1.2+cu.12.6.torch.2.7 , 2.1.2+cu.12.6.torch.2.6 transformer-engine-torch 2.16.0+cu.12.6.torch.2.12 , 2.16.0+cu.12.6.torch.2.11 , 2.16.0+cu.12.6.torch.2.10 , 2.16.0+cu.12.6.torch.2.9 , 2.16.0+cu.12.6.torch.2.8 , 2.16.0+cu.12.6.torch.2.7 , 2.16.0+cu.12.6.torch.2.6 vllm 0.10.0+cu126 , 0.9.2+cu126 , 0.9.1+cu126