Model compiler installation and configuration#
The following describes installing and setting up the model compiler & optimizer and neural net models for Linux.
Prerequisites#
Linux
Ubuntu 20.04 / 22.04
CentOS 7 / 8
RHEL 8 / 9
MIVisionX installed as described in Installation
Install Linux Packages
Ubuntu
sudo apt-get -y install python3 python3-pip protobuf-compiler libprotoc-dev
CentOS/RHEL
sudo yum -y python3-devel python3-pip protobuf python3-protobuf
Install PIP3 and Python Packages
sudo pip3 install future==0.18.2 pytz==2022.1 numpy==1.23.0
Note
MIVisionX installs model compiler scripts at /opt/rocm/libexec/mivisionx/model_compiler/python/
Setting up neural networks#
The following lists the requirements and steps to configure supported neural nets for use with model compiler.
Caffe#
protobuf
google
sudo pip3 install google==3.0.0 protobuf==3.12.4
ONNX#
protobuf
onnx
sudo pip3 install protobuf==3.12.4 onnx==1.12.0
Note
ONNX Models are available at ONNX Model Zoo
NNEF#
nnef-parser - Build the nnef python module
git clone -b nnef-v1.0.0 https://github.com/KhronosGroup/NNEF-Tools.git cd NNEF-Tools/parser/cpp mkdir -p build && cd build cmake ../ make sudo make install cd ../../../python sudo python3 setup.py install
Note
NNEF models are available at NNEF Model Zoo