I think the ARM64 builds from PyTorch.org don’t include CUDA, so you would need the libraries from NVIDIA
PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums 18
(I also built packages for it 2, but you’ll likely be happier with the ones from NVIDIA.)
Best regards
Thomas