Install Pytorch with cuda on Jetson Orin nano Devloper Kit

Autonomous Machines Jetson & Embedded Systems Jetson Orin Nano
,
6

Hi,

Please try the below command:

$ wget https://developer.download.nvidia.com/compute/redist/jp/v60/pytorch/torch-2.4.0a0+07cecf4168.nv24.05.14710581-cp310-cp310-linux_aarch64.whl -O torch-2.4.0a0+07cecf4168.nv24.05.14710581-cp310-cp310-linux_aarch64.whl
$ pip3 install torch-2.4.0a0+07cecf4168.nv24.05.14710581-cp310-cp310-linux_aarch64.whl

If the error keeps showing, please share the output of the below command with us.

$ md5sum torch-2.4.0a0+07cecf4168.nv24.05.14710581-cp310-cp310-linux_aarch64.whl
$ apt show nvidia-jetpack

Thanks.