William-wAng618
commented
3 weeks ago Did you solve this question?
Open agm-eratosth opened 2 months ago
William-wAng618
commented
3 weeks ago Did you solve this question?
🚀 The feature
torchvision provides prebuilt binaries for aarch64 + cuda on linux for pip and conda
Motivation, pitch
pytorch now provides aarch64 cuda prebuilt binaries in 2.4.0 (https://github.com/pytorch/pytorch/issues/134117) but currently torchvision 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 torchvision from source, which is not ideal for users of this library since it requires lots of work.
Additional context
Some relevant discussion here: https://discuss.pytorch.org/t/cuda-pytorch-package-manager-on-gh200/208582