onnx 使用初体验

来源:互联网 发布:fanuc机器人手动编程 编辑:程序博客网 时间:2024/05/17 05:07

需要使用 pytorch, caffe2, coreml, onnx.
由于 caffe2 只支持 python2, 使用 python2.7.14来作为虚拟环境来编译 pytorch的时候,报错:

clang: error: invalid deployment target for -stdlib=libc++ (requires OS X 10.7 or later)clang: error: invalid deployment target for -stdlib=libc++ (requires OS X 10.7 or later)error: command '/usr/bin/clang' failed with exit status 1

clang error 可以添加
故选择使用 anaconda2, 依次安装依赖。

需要注意的是,由于已经有 anaconda3 的环境,在虚拟环境中使用 conda 命令的时候使用的依然是 anaconda3 的环境。如果在虚拟环境中需要使用 conda的时候,带一下anaconda2的路径即可。
各种路径冲突,没找到更好的办法,只能在 conda info 中先用 anaconda2的路径覆盖 anaconda3 的路径。

1. 安装 onnx

github tutorial.

conda_path/conda install -c ezyang onnx

然后,检验是否安装成功:

python -c 'import onnx'ImportError: No module named onnx

好的,安装失败。

在覆盖了conda的路径之后,使用 conda 的方法来安装 onnx.

conda install -c ezyang onnx

由于在下载 mkl的时候速度太慢了,可以前往 anaconda cloud 手动下载安装 mkl.

python -c 'import onnx'

无报错提示,安装成功。

2. 安装 onnx-coreml

github tutorial.

pip install -U onnx-coreml

顺利安装成功。

3. 安装 pytorch

如果需要onnx相关依赖,需要从源码安装。
github tutorial.

Install optional dependencies
On mac OS
export CMAKE_PREFIX_PATH=[anaconda root directory]
conda install numpy pyyaml setuptools cmake cffi

Get the PyTorch source
git clone –recursive https://github.com/pytorch/pytorch

cd pytorch
MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install

在电脑风扇封装转之后,安装完成。

原创粉丝点击