Nightly CUDA 12.4 builds of PyTorch 2.4 and 2.5 for aarch64 with CUDA support can be installed via pip.
Once 2.4 is released at some point in July 2024, it should be possible to install from the non-nightly channel, hopefully for CUDA 11.8 and 12.1, and hopefully via conda.
These instructions are added as a temporary work around after a user enquiry
Nightly CUDA 12.4 builds of PyTorch 2.4 and 2.5 for aarch64 with CUDA support can be installed via pip.
Once 2.4 is released at some point in July 2024, it should be possible to install from the non-nightly channel, hopefully for CUDA 11.8 and 12.1, and hopefully via conda.
These instructions are added as a temporary work around after a user enquiry
Related to #199