Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Prebuilt binaries of torch.audio for aarch64 cuda #3827

Open
chulkilee opened this issue Aug 26, 2024 · 1 comment
Open

Prebuilt binaries of torch.audio for aarch64 cuda #3827

chulkilee opened this issue Aug 26, 2024 · 1 comment

Comments

@chulkilee
Copy link

chulkilee commented Aug 26, 2024

🚀 The feature

torch.audio provides prebuilt binaries for aarch64 + cuda on linux

Motivation, pitch

pytorch now provides aarch64 cuda prebuilt binaries in 2.4.0 (pytorch/pytorch#134117) but currently torch.audio doesn't provide corresponding prebuilt binaries. It makes very hard to adopt torch on aarch64 + cuda environment, which is often much cheaper in cloud providers.

Alternatives

Build torch.audio from source, which is not ideal for users of this library since it requires lots of works.

Additional context

No response

@ptrblck
Copy link
Collaborator

ptrblck commented Sep 12, 2024

Nightly binaries are already available for ARM+CUDA. The PyTorch 2.4.0 release only added the torch wheels without torchvision and torchaudio as the latter two builds were created late in the release cycle and thus to avoid last minute churn.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

5 participants
@chulkilee @ptrblck and others