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 GPUCUDA 11.3 + CuDNN 8 |
Windows CPUโฅ Visual Studio 2015.2 |
Windows GPUโฅ CUDA 8.0 + CuDNN 5.1 |
|---|---|---|---|---|
 |
 |
Usage
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
- How to quantize neural network models
- 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.
Installation
Get the N2D2 Source
git clone --recursive [email protected]:CEA-LIST/N2D2.git
Specify the recursive option is required as it will download the PyBind submodule.
Install Dependencies
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.
Build on Linux
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.
Docker Image
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.
Contributing
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.
License
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.