search

CEA-LIST / N2D2

N2D2 is an open source CAD framework for Deep Neural Network simulation and full DNN-based applications building.
Other
146 stars 36 forks source link

ONNX not available when using dockerized N2D2 #103

Closed e-dupuis closed 2 years ago

e-dupuis commented 2 years ago

Using the Dockerfile provided

[...]
export_ONNX_float32/stimuli_stats processing 50000 stimuli
sh: 1: dot: not found
Exporting Test dataset to "export_ONNX_float32/stimuli"....................
Error: "ONNX" export is not available (additional modules may be required)

steps to reproduce:

docker build -t n2d2 .
docker run -it --rm n2d2 n2d2 MobileNet_ONNX.ini -seed 1 -w /dev/null -export ONNX
e-dupuis commented 2 years ago

Adding The two following installs in the Dockerfile fix the issue: protobuf-compiler libprotoc-dev 

export_ONNX_float32/stimuli_stats processing 50000 stimuli
sh: 1: dot: not found
Exporting Test dataset to "export_ONNX_float32/stimuli"....................
Generating ONNX export to "export_ONNX_float32":
-> Generating network
Fuse Padding...
sh: 1: dot: not found
Notice: Unused section  in INI file
Done!
olivierbichler-cea commented 2 years ago

Thanks for this fix, it has been merged!