My Jetpack version is 5.1.4 and however I can only find this link https://developer.download.nvidia.cn/compute/redist/jp/v511/pytorch/torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl
and I got this error when installing it
ERROR: pip’s dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
torchaudio 2.0.2 requires torch==2.0.1, but you have torch 2.0.0+nv23.5 which is incompatible.
torchvision 0.15.2 requires torch==2.0.1, but you have torch 2.0.0+nv23.5 which is incompatible.
Where can I find the link for 2.1.0a?
In the end, python will need to find the cuda like this: python3 -c “import torch; print(torch.cuda.is_available()”
Or is there another way to utilize CUDA in python?