Skip to content

Request: Provide +cpu wheels for ARM64 (aarch64) in official TorchVision releases #9197

@DHolmanCoding

Description

@DHolmanCoding

🚀 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

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions