ONNX源码安装

ONNX是facebook提出的一个 Open Neural Network Exchange协议,可以让训练好的模型在不一样的框架间进行交互。python

ONNX的安装相对来讲不是特别麻烦,麻烦的是其依赖库的安装。ONNX依赖于pybind11。git

首先依赖库的安装github

sudo pip install pytest #pytest sudo pip install numpy #numpy sudo pip install scipy #scipy 

下载pybind11源码python3.x

git clone https://github.com/pybind/pybind11.git 

 若是你须要eigen库的话,还须要先安装eigen,这里我选择的是无eigen的方式bash

新建一个编译出来的文件夹框架

mkdir build cd build cmake .. make -j4 sudo make install 

 这里有个问题,若是你同时存在python2.x和python3.x版本的话,默认选择的是python3.x版本的,可是我如今的整个环境是python2.7.12,因此就必须禁用掉Python3.x,你们都说禁用,其实就是让自动寻找方式找不到python3.x。python2.7

采用的方式是在/usr/bin中全部找到的python3.x相关的文件都移动到另外一个文件夹内,而后开始编译ui

装好了pybind11以后(源码版本编译ONNX必须手动编译pybind11,使用pip install pybind11并无论用)spa

编译安装onnxcode

 

git clone https://github.com/onnx/onnx.git cd ./onnx 
python setup.py build sudo python setup.py install 

这里颇有可能会出现cmake提示 Protobuf compiler not found  的问题,所以须要安装合适的protobuf编译器

 

sudo apt-get install  libprotobuf-dev protobuf-compiler
相关文章
相关标签/搜索