-
Notifications
You must be signed in to change notification settings - Fork 7.1k
Description
🚀 The feature
Please publish ARM64 (aarch64) CPU-only +cpu wheels for TorchVision to the official PyTorch CPU index at
https://download.pytorch.org/whl/torchvision/
so they mirror the existing x86_64 +cpu artifacts (and match the torch CPU-only availability, which already includes aarch64).
Concretely:
Build and upload torchvision==+cpu wheels for manylinux aarch64 (and macOS arm64 if you use the +cpu channel there).
Keep version/ABI pairing aligned with the corresponding torch==+cpu wheels.
Document the install matrix (x86_64 & aarch64 parity) in the install guide.
This makes the Torch/TorchVision packaging matrix symmetric across architectures and channels, and avoids special-casing ARM in dependency managers.
Motivation, pitch
I’m building lightweight CPU-only deployments on ARM64 devices (aarch64). The goal is to ship small images without CUDA/NVIDIA baggage. Today there’s an asymmetry:
torch provides aarch64 +cpu wheels on the CPU index.
torchvision does not provide aarch64 +cpu wheels on that index; ARM64 wheels exist only as plain (no +cpu) on PyPI.
Alternatives
No response
Additional context
No response