What are Wheel Variants?
- Wheel variants are a mechanism for publishing platform-dependent Python wheels and selecting the most suitable package variant for a given platform.
- This approach removes the need for manual selection of local identifiers in PyTorch packaging and greatly enhances the user experience when installing PyTorch.
- For technical details and ongoing discussion, please see PEP-817 and associated discourse thread
What is currently supported?
The following variants are supported for PyTorch 2.10.0:
- CUDA 12.6, 12.8 and 13.0: Linux x86, Linux aarch64, Windows x86
- ROCm 7.0 and 7.1
- XPU: Windows x86 and Linux x86
- CPU only: Linux, Linux aarch64, MacOS (including M1/M2), Windows (CPU only)
How to test it?
You can start using Wheel Variants today by installing the PyTorch 2.10.0 release with the variant-enabled uv installer, which will automatically select the best PyTorch build for your machine’s hardware.
Using Linux/MacOS:
curl -LsSf https://astral.sh/uv/install.sh | INSTALLER_DOWNLOAD_URL=https://wheelnext.astral.sh/v0.0.3 sh
uv pip install torch torchvision
Using Windows:
powershell -ExecutionPolicy Bypass -c "$env:INSTALLER_DOWNLOAD_URL='https://wheelnext.astral.sh/v0.0.3'; irm https://astral.sh/uv/install.ps1 | iex"
uv pip install torch torchvision
If you have any questions or feedback, please feel free to reach out.
Cheers,
Team PyTorch