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 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.
Operating system | 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 |
|
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 all wheels with the clang compiler to prevent issues with the ARM NEON registers and the GNU compiler.
For instance #61110 and #65673.
You should also use the clang compiler (version 8) if you want to compile 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