ay-yay / n2d2 Goto Github PK

N2D2 (for Neural Network Design & Deployment) is CEA LIST's CAD framework for designing and simulating Deep Neural Network (DNN), and building full DNN-based applications on embedded platforms. N2D2 is developped along with industrial and academic partners and is open source.

Docs	Linux CPU ≥ GCC 4.4.7	Linux GPU CUDA 11.3 + CuDNN 8	Windows CPU ≥ Visual Studio 2015.2	Windows GPU ≥ CUDA 8.0 + CuDNN 5.1

You can discover how to use the framework by visiting our online documentation.

• General presentation of the framework
• How to compile N2D2 and perform simulations
• How to write neural network models
  • With the INI interface
  • With the Python API
• How to quantize neural network models
  • With Post-Training Quantization
  • With Quantization-Aware Training
• How to export neural network models
• Tutorials

The N2D2 executables and application examples are located in the exec/ directory.

You can find examples of the N2D2 Python API here.

git clone --recursive [email protected]:CEA-LIST/N2D2.git

Specify the recursive option is required as it will download the PyBind submodule.

The only mandatory dependencies for N2D2 are OpenCV and Gnuplot.

Moreover, the NVIDIA CUDA and CuDNN libraries are required to enable GPU-acceleration. We highly recommend to use a CUDA version higher than 10 with a CuDNN version higher than 7.

If you want to disable CUDA support, export the environment variable N2D2_NO_CUDA=1.

To compile N2D2 on Linux, please go to the root of the project and run the following:

mkdir build
cd build
cmake .. && make

You should have the n2d2 executable in build/bin after the compilation.

To install the Python API in your python environment, follow the tutorial on our doc.

You can also pull a pre-built docker image from Docker Hub and run it with docker

docker pull cealist/n2d2
docker run --gpus all cealist/n2d2:latest

Another possibility is to build N2D2 from the Dockerfile. It is supplied to build images with CUDA 10.2 support and CuDNN 8.

If you would like to contribute to the N2D2 project, we’re happy to have your help! Everyone is welcome to contribute code via pull requests, to file issues on GitHub, to help people asking for help, fix bugs that people have filed, to add to our documentation, or to help out in any other way.

We grant commit access (which includes full rights to the issue database, such as being able to edit labels) to people who have gained our trust and demonstrated a commitment to N2D2.
For more details see our contribution guidelines.

N2D2 is released under the CeCILL-C license, a free software license adapted to both international and French legal matters that is fully compatible with the FSF's GNU/LGPL license.