Qengineering / Jetson-Nano-image

Jetson Nano image with deep learning frameworks
https://qengineering.eu/install-tensorflow-2.4.0-on-jetson-nano.html
BSD 3-Clause "New" or "Revised" License
118 stars 23 forks source link
cuda deep-learning jetson-nano mnn ncnn opencv pytorch sd-card-image team-viewer tegra tensorflow torch torchvision

Jetson Nano DNN image

output image

A Jetson Nano image with OpenCV, TensorFlow and Pytorch

License


Installation.


Tips.


Pre-installed frameworks.

Tensorflow 2.5 and above require CUDA 11. CUDA version 11 cannot be installed on a Jetson Nano due to incompatibility between the GPU and low-level software at this time, hence Tensorflow 2.4.1. Only when NVIDIA releases a JetPack with CUDA 11 will we be able to upgrade Tensorflow.

output image

output image


OpenCV + TensorFlow.

Importing both TensorFlow and OpenCV in Python can throw the error: cannot allocate memory in static TLS block.
This behaviour only occurs on an aarch64 system and is caused by the OpenMP memory requirements not being met.
For more information, see GitHub ticket #14884.

output image

There are a few solutions. The easiest is to import OpenCV at the beginning, as shown above.
The other is disabling OpenMP by setting the -DBUILD_OPENMP and -DWITH_OPENMP flags OFF.
Where possible, OpenCV will now use the default pthread or the TBB engine for parallelization.
We don't recommend it. Not all OpenCV algorithms automatically switch to pthread.
Our advice is to import OpenCV into Python first before anything else.


paypal