Data

Tools

Indexes

Applications

Experiments

Awesome

An open API service indexing awesome lists of open source software.

https://github.com/qengineering/pytorch-jetson-nano

PyTorch installation wheels for Jetson Nano
https://github.com/qengineering/pytorch-jetson-nano

aarch64 cuda cuda-10 cudnn cudnn8 jetson-nano jetson-tx2 jetson-xavier libtorch python-wheel python3 pytorch pytorch-installation torch torchvision wheel

Last synced: 3 months ago
JSON representation

PyTorch installation wheels for Jetson Nano

Awesome Lists containing this project

README 

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

![output image]( https://qengineering.eu/images/PyTorch_icon.webp )




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](https://github.com/Qengineering/Jetson-Nano-Ubuntu-20-image). We supply the wheels for this version.



-----------------



Wheel: the installation wheel torch-_version_-cp**xx**-cp**xx**-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](https://drive.google.com/file/d/1e9FDGt2zGS5C5Pms7wzHYRb0HuupngK1/view?usp=share_link)
[Vision](https://drive.google.com/file/d/19UbYsKHhKnyeJ12VPUwcSvoxJaX7jQZ2/view?usp=share_link)
[LibTorch](https://drive.google.com/file/d/1k8nDUFUI_5_07MKkZTJzK4o1TEtks9mQ/view?usp=share_link)
[Guide](https://qengineering.eu/install-pytorch-on-jetson-nano.html) | [Wheel](https://drive.google.com/file/d/1MnVB7I4N8iVDAkogJO76CiQ2KRbyXH_e/view?usp=sharing)
[Vision](https://drive.google.com/file/d/11DPKcWzLjZa5kRXRodRJ3t9md0EMydhj/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1t0MM1Bec2XIIKK8PhQbEDOrM1Z2Xym5-/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-jetson-nano.html) | [Wheel](https://drive.google.com/file/d/1AQQuBS9skNk1mgZXMp0FmTIwjuxc81WY/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1BaBhpAizP33SV_34-l3es9MOEFhhS1i2/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1OSWB_Wv7rghpiBI3V9Rvj0ZR6bRcAZsY/view?usp=sharing)
[TorchText](https://ln5.sync.com/dl/3a2821c60/emd4zfyb-r5wvmdk4-2jkhf29x-7d2c7akt)
[Guide](https://qengineering.eu/install-pytorch-on-jetson-nano.html)
[TorchText gist](https://gist.github.com/Qengineering/0aef6533ec7cc82cddc6fbe70e907f33)|||

| **Jetson Nano JetPack 4.6**
(Python 3.6) | | | | |[Wheel](https://drive.google.com/file/d/1TqC6_2cwqiYacjoLhLgrZoap6-sVL2sd/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1C7y6VSIBkmL2RQnVy8xF9cAnrrpJiJ-K/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1izv6kmcnqXk9i7-Ey-vldjC-CGfHOGCl/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-jetson-nano.html)| [Wheel](https://drive.google.com/file/d/1wzIDZEJ9oo62_H2oL7fYTp5_-NffCXzt/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1Q2NKBs2mqkk5puFmOX_pF40yp7t-eZ32/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1E4Hfz1cj6bwGz8a72OS5uH3SnlvRyrOi/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-jetson-nano.html) | [Wheel](https://drive.google.com/file/d/1-XmTOEN0z1_-VVCI3DPwmcdC-eLT_-n3/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1BdvXkwUGGTTamM17Io4kkjIT6zgvf4BJ/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-jetson-nano.html) | [Wheel](https://drive.google.com/file/d/1aWuKu8eqkZwVzFFvguVuwkj0zdCir9qX/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1aWuKu8eqkZwVzFFvguVuwkj0zdCir9qX/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-jetson-nano.html) |

| Raspberry Pi 64-bit Bullseye
(Python 3.9) | [Wheel](https://drive.google.com/file/d/1mPlhwM47Ub3SwQyufgFj3JJ9oB_wrU5D/view?usp=sharing)
[Vision](https://drive.google.com/file/d/14AJEjzi3upgguI2RdBDEGj0jzc7ItvUC/view?usp=drive_link)
[LibTorch](https://drive.google.com/file/d/1N67Pqb0I1e-k86BQIUr1gSKPqfnpaKu3/view?usp=drive_link)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/1uLkZzUdx3LiJC-Sy_ofTACfHgFprumSg/view?usp=share_link)
[Vision](https://drive.google.com/file/d/1AhbkLqKd8EZO2pZV_g9aFZGHZo2Ubc3O/view?usp=share_link)
[LibTorch](https://drive.google.com/file/d/10HvaVNJ4fVBNuGeI3GNFz6e-xiGK9v5P/view?usp=share_link)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/1A2Lc-7y3-ATauRa4FntiYGH90b1GjAzx/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1VBegXwqS-A4MVNlHt8Y_0cu697NrAtf7/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/10VFc7XMi9Dx9beZoAoi2lFVZe6a8KMAs/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html)  | [Wheel](https://drive.google.com/file/d/1ilCdwQX7bq72OW2WF26Og90OpqFX5g_-/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1oDsJEHoVNEXe53S9f1zEzx9UZCFWbExh/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1-kTx72-6eePrXVjFtJN-3ntu1DCI8FuD/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/10zJmU1f7PoxN8ARZol4mtbwok4G23XKZ/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1nX4uSoubwq8NB9k-2DF6LbPyiqwocgEW/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1z58SN6Inh3ka_jFs97LefBuvfGq9o1to/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/1p9Sp5YmCJwIkmQIWkpuIkE4NgeYPOr-U/view?usp=sharing)
[Vision](https://drive.google.com/file/d/13fKq1V-zLbLBWPw3zP4EZxyMXyOiJWPi/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1My9PPoa9oYv7ZaPdsoXq4FyqRCHHA_4c/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/1E4bP9NAG5pDSXGWYPGsJ5uzFBq47VN14/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1mqNqUC1t0MekF5h1WCEqt3soKWhc6Py4/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1PrqFOPHxrbLGEMuxBTxYY-Ubjk-l88_Z/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) |  ||||||||

| Raspberry Pi 64-bit Buster
(Python 3.7) | | [Wheel](https://drive.google.com/file/d/1FQw6NgT9k3paiAhEbq0ibkVSIGNLAyro/view?usp=share_link)
[Vision](https://drive.google.com/file/d/1m6wEbl_jJGaQL-vg4zkNfc9TNwl2_J4o/view?usp=share_link)
[LibTorch](https://drive.google.com/file/d/1pt5LGdYDL5KJjcKcbjVjrCYZTJVOOlGT/view?usp=share_link)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html)  | [Wheel](https://drive.google.com/file/d/1q0NJABGqXIHnOPbFUNrMZvNvrzNAWklq/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1b6hfBZ4US9ukMhd5f0AKDxhHtKSofbNG/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1rJPFWkm3WWw1Vx8gZWf3hTCZ0v8yFn_S/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/1gAxP9q94pMeHQ1XOvLHqjEcmgyxjlY_R/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1btEPxKYfL6OP1Xaedly-ttfPTH-mu4fp/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1sj2V0RONtSLHHjmQ904U82YjdV0TM5bf/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html)|[Wheel](https://drive.google.com/file/d/1y-X1UGC43xUfnycJcuDp7tlyFi8QHxvs/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1uQgjrucNSfpliGBWeKUOuo4XGWiBaxQx/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1nC_gWNjFirZz74wV4z7t2un4ktCtlRpI/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html)| [Wheel](https://drive.google.com/file/d/18KztAf3W_kqFYP2f9uqLj3HBdhoMvVe4/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1XyYG6hHh7QYaLrvE-mByvhNK_0rLOd0G/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1yhFMxngRq_RY2DPWwqZ8wwNmzBBjzrHR/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/1JF3x556B1ZPe7rp3ogyZiCJPCDsu57RY/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1isEX4w4oegkYpv7WalvYPkTpPCcI4CUR/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1kpAb7qIGD0ICy8HYHDGMiY0Yj-K-eGAP/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/1o2PjvvKkHpYx7QMfMLUoNck7ZiE_TGgR/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1H3YLkOicAN78tBLAmzCx2lt_WBjjmFCz/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1XpdE-AG4QMz1HE8bDdpiKw1Fpl3Em0kK/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) ||||||||

| Raspberry Pi Ubuntu 18.04
(Python 3.6) | | | | | |  | [Wheel](https://drive.google.com/file/d/18WNCbSLeQCZUJuXf-0yhzopDR6p4SkYn/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1QhUPbWvotJQyUfF8ZxYgFHDPHLQagC3U/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1ORi9-9q82CoxdL7TmnZ7-e3qFzqJ-DAD/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/1ciFrD5BekDd1ubVOCwK-oGfSUJwF8_WA/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1hpJ_5esV4VAHy8Bq8ZJ8SuD2w8M8Dw3m/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1wbJTRxrFHZ6UOAzZ8w8RU0rYdyWibDNS/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) |  

| Raspberry Pi Ubuntu 20.04
(Python 3.8) | | | | | [Wheel](https://drive.google.com/file/d/1W7hhNG_HJUyY0sk1dBikbSRyAJr_Nyc4/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1SmWRjEyizWT7iumXUZATZuBSfQt0gJ8o/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1iV2c7fENEG8L6-9l2CL_0o5Z1kJ53YB-/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) | [Wheel](https://drive.google.com/file/d/12GH3r-uaRxUIzcHHWcfXtisXk71OZbF0/view?usp=sharing)
[Vision](https://drive.google.com/file/d/1VgEinJIu5e38h1aKaFSn8A1F1MaNeM5g/view?usp=sharing)
[LibTorch](https://drive.google.com/file/d/1bC34mTPJbtHZX_c675dUdmFHo8d8X9jT/view?usp=sharing)
[Guide](https://qengineering.eu/install-pytorch-on-raspberry-pi-4.html) ||||



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](https://github.com/pytorch/pytorch/issues/61110) and [#65673](https://github.com/pytorch/pytorch/issues/65673).



![output image]( https://qengineering.eu/images/ClangRPi.png )



![output image]( https://qengineering.eu/images/Torch_1_10_0_Jetson_Succes_GitHub.webp )


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



-----------------



![output image](https://qengineering.eu/images/SDcard32GB_smallJetson.jpg) Find PyTorch and TorchVision with other frameworks and deep-learning examples on our [SD-image](https://github.com/Qengineering/Jetson-Nano-imagee)



-----------------



[![paypal](https://qengineering.eu/images/TipJarSmall4.png)](https://www.paypal.com/cgi-bin/webscr?cmd=_s-xclick&hosted_button_id=CPZTM5BB3FCYL)