How to install PyTorch on Jetson Orin?

[danielburdisso49]

I am trying to install the correct PyTorch for a Jetson Orin given the following information about the system which is the output from nvidia-smi.

NVIDIA-SMI 540.3.0 Driver Version: N/A CUDA Version: 12.2

The Orin is using Ubuntu version 22.04 with JetPack version R36 revision 3.0. The python version is 3.10.12.

I have tried the steps on the NVIDIA page for installing on an NVIDIA platform located here: Installing PyTorch for Jetson Platform - NVIDIA Docs

I have also tried the installation steps in the PyTorch website itself including several of the older PyTorch versions it has listed for older CUDA versions.

None of these have worked as I am still getting false when running ‘torch.cuda.is_available()’ in Python.

Has anyone successfully installed PyTorch on a Jetson Orin with JetPack R36.3 (CUDA 12.2)? If so, which PyTorch and torchvision versions worked for you, and how did you install them?

Thanks in advance for your help!

[whitesscott]

The one on this page may work https://pypi.jetson-ai-lab.io/jp6/cu126

[AastaLLL]

Hi,

Do you have a dependency on JetPack 6.0?
If not, it’s recommended to upgrade to the latest JetPack 6.2.1 (CUDA 12.6+R36.4.4).

We have PyTorch prebuilt for JetPack 6.2 in the link below:

jp6/cu126 index

Thanks.