PyTorch-Jetson-Nano

PyTorch wheels for Jetson Nano, TX2, (AGX) Xavier

Find your operating system and Pytorch version in the table below. Follow the instructions in the provided guide.
The Jetson Nano wheels supports CUDA 10.2, cuDNN 8.0 and NEON.

---

PyTorch 2.0

Pytorch 2.0 and above uses CUDA 11. The Jetson Nano has CUDA 10.2.
Due to low-level GPU incompatibility, installing CUDA 11 on your Nano is impossible.
Pytorch 2.0 can only be installed on Jetson family members using a JetPack 5.0 or higher, such as the Jetson Nano Orion.
Unfortunately, it does not appear that this version will also be available for the Jetson Nano soon.

---

PyTorch 1.13.0, 1.12.0 and 1.11.0

PyTorch version 1.11 and above requires Python 3.7.
Since JetPack 4.6 has Python 3.6, you cannot install PyTorch 1.11.0 on a Jetson Nano.

It looks like Nvidia has no plans to release JetPack 5.0 for the Jetson Nano for now. It’s only available for the Xavier series.
You can use the current experimental version of Jetson Nano with Ubuntu 20.04. We supply the wheels for this version.

---

Wheel: the installation wheel torch-version-cpxx-cpxx-linux_aarch64.whl (xx is the used python version)
Vision: the accompanying torchvision.
LibTorch: the C++ API for those who like to program. (The aarch64 version of libtorch-cxx11-abi-shared-with-deps-1.10.1+cu102.zip)
Guide: link to the installation tutorial.

Roadmap.

Operating system	PyTorch 2.0.0	PyTorch 1.13.0	PyTorch 1.12.0	PyTorch 1.11.0	PyTorch 1.10.0	PyTorch 1.9.0	PyTorch 1.8.0	PyTorch 1.7.0
Jetson Nano Ubuntu 20.04 (Python 3.8)		Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch TorchText Guide TorchText gist
Jetson Nano JetPack 4.6 (Python 3.6)					Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision Guide	Wheel Vision Guide
Raspberry Pi 64-bit Bullseye (Python 3.9)	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide
Raspberry Pi 64-bit Buster (Python 3.7)		Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide
Raspberry Pi Ubuntu 18.04 (Python 3.6)							Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide
Raspberry Pi Ubuntu 20.04 (Python 3.8)					Wheel Vision LibTorch Guide	Wheel Vision LibTorch Guide

We compiled the Pytorch 1.x versions with the clang compiler to prevent issues with the ARM NEON registers and the GNU compiler. For instance #61110 and #65673.




Pytorch 2.0 was build with the GNU compiler.

:heavy_exclamation_mark: C++ programmers please note

You should also use the clang compiler (version 8) if you want to compile Pytorch 1.x versions C++ code yourself.
The GNU GCC compiler will give you ‘no expression errors’.

# set clang compiler at the command line
$ export CC=clang
$ export CXX=clang++

Don’t worry if you plan to use Python. It only applies to C++ users.

---

Find PyTorch and TorchVision with other frameworks and deep-learning examples on our SD-image

---