Install Pytorch with cuda on Jetson Orin nano Devloper Kit

dusty_nv · July 8, 2024, 6:51pm

For other versions of Python than the default, you will need to build PyTorch from source like shown in this topic, or using jetson-containers:

github.com

dusty-nv/jetson-containers/blob/master/docs/build.md#changing-versions

# Building Containers

The [`jetson-containers build`](/jetson-containers) command is a proxy launcher for [`jetson_containers/build.py`](/jetson_containers/build.py).  It can be run from any working directory after you clone the repo and run the installer from the [System Setup](/docs/setup.md) (you should also probably mount additional storage when building containers)

To list the packages available to build for your version of JetPack/L4T, you can use `--list-packages` and `--show-packages`:

```bash
$ jetson-containers build --list-packages       # list all packages
$ jetson-containers build --show-packages       # show all package metadata
$ jetson-containers build --show-packages ros*  # show all the ros packages
```

To build a container that includes one or more packages:

```bash
$ jetson-containers build pytorch               # build a container with just PyTorch
$ jetson-containers build pytorch jupyterlab    # build container with PyTorch and JupyterLab
```

The builder will chain together the Dockerfiles of each of packages specified, and use the result of one build stage as the base image of the next.  The initial base image defaults to one built for your version of JetPack, but can be changed by specifying the [`--base` argument](#changing-the-base-image).

This file has been truncated. show original

Topic		Replies	Views
PyTorch for Jetson Orin Jetpack 5.1 installation issue Jetson AGX Orin pytorch	12	1416	January 23, 2024
Pytorch installed on l4t-jetpack:r35.4.1 container on Jetson Orin Nano (JetPack 6.0 Developer Kit) fails to recognize CUDA Jetson Orin Nano cuda , docker , pytorch , python , containers	2	159	October 22, 2024
Pytorch compatibility issues (torch 2.0.0+nv23.5 && torchvision 0.15.1) Jetson Orin NX pytorch	10	16763	June 13, 2023
Unable to get Pytorch vision & audio working on Jetpack 6.2 Jetson AGX Orin pytorch	10	92	June 11, 2025
JetPack 6.0 – Missing PyTorch + CUDA Support for Jetson Inference (Need Official Wheel or Build Instructions) Jetson Orin Nano pytorch	15	126	May 26, 2025
Cant install Pytorch on JetsonNano P3450 Jetson Nano pytorch	21	2509	August 16, 2023
Cannot get Torch met CUDA to work on Jetson Orin Jetson Orin NX pytorch	10	2225	February 26, 2024
Issue to install pytorch on Jetson Orin platform Jetson Orin NX jetson-inference , pytorch	10	5981	June 7, 2023
Unable to use GPU with pytorch yolov5 on jetson nano Jetson Nano cuda , pytorch	2	2709	October 20, 2022
Hello AI World for Jetpack 6.0 DP - Pytorch 2.1.0 Installed, Torchvision Did Not Install Jetson Orin Nano pytorch	24	1855	January 15, 2024

Install Pytorch with cuda on Jetson Orin nano Devloper Kit

Related topics