Find your operating system and Pytorch version in the table below. Follow the instructions in the provided guide.
There is no Raspberry Pi 32-bit wheel available due to unsupported libraries.
The Jetson Nano wheels support CUDA 10.2, cuDNN 8.0 and NEON. They can also be used on the (AGX) Xavier.
If you have a Bookworm OS on your Rpi4 or Rpi5, you can use the wheel provided by PyTorch.
The wheels here do not support Bookworm with its Python version 3.11.
Please see our site: install pytorch on raspberry pi 5.
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+cpu.zip)
Guide: link to the installation tutorial.
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 |
---|---|---|---|---|---|---|---|---|
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 |
||||||
Jetson Nano JetPack 4.6 (Python 3.6) |
Wheel Vision LibTorch Guide |
Wheel Vision LibTorch Guide |
Wheel Vision Guide |
Wheel Vision Guide |
||||
Jetson Nano Ubuntu 20.04 (Python 3.8) |
Wheel Vision LibTorch Guide |
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.
You should also use the clang compiler 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