Earlier I found a discussion here https://discuss.pytorch.org/t/conda-pytorch-cuda-12-for-arm64-and-grace-hopper/203285, I wonder when the official ARM+CUDA PyTorch support will be on live? At least at the moment I still can’t get it from conda install.
ARM+CUDA torch pip wheels are supported as nightly binaries (including vision and audio).