Pytorch on Jetson nano Jetpack 4.4 R32.4.4

[juan.caceres]

I’m trying to install pytorch with pip instructions from pytorch.org in jetpack 4.4 R32.4.4 with cuda 10.2 but the procedure fail also the make fail somebody achieve this with success?

[AastaLLL]

Hi,

You can find the package for Jetson on the following topic:

PyTorch for Jetson

Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the pre-built l4t-pytorch and l4t-ml container images and Dockerfiles. PyTorch pip wheels JetPack 5

PyTo…

Or our NGC container: l4t-pytorch or l4t-ml.

Thanks.

[juan.caceres]

This one requires Cuda 10.0 and jetpack 4.4 comes with 10.2 some solution?

[AastaLLL]

Hi,

You should be able to find the corresponding package for each JetPack version.

For example:

PyTorch v1.7.0

• JetPack 4.4 (L4T R32.4.3) / JetPack 4.4.1 (L4T R32.4.4)
  • Python 3.6 - torch-1.7.0-cp36-cp36m-linux_aarch64.whl

Thanks.