PyTorch wheels for Raspberry Pi 4

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.

Bookworm.

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.

Roadmap.

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 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