使用pytorch输出onnx模型并转换成openvino的IR模型

经过好几天的试错,终于成功转换出模型了,出错主要问题是各种包版本不一致的问题。

安装pytorch/torchvision

安装环境和包版本:
Windows 10
torch:1.5.1;
torchvision:0.6.1;
python:3.7.6
onnx:1.7.0
openVino:Anaconda3-2020.02-Windows-x86_64.exe
tensorflow>=1.2.0,<2.0.0
mxnet>=1.0.0,<=1.3.1
networkx=2.0
numpy=1.19.0
protobuf3.6.1
test-generator0.1.1
defusedxml>=0.5.0

anaconda3环境安装:
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
安装openVino时,默认会按照如下版本安装
tensorflow>=1.2.0,<2.0.0
mxnet>=1.0.0,<=1.3.1
networkx>=1.11
numpy>=1.12.0
protobuf3.6.1
onnx>=1.1.2
test-generator0.1.1
defusedxml>=0.5.0
所以,如果没有设置最高版本都会按最新版本安装,但是会导致版本不一致,其中networkx的版本默认安装是2.4,但生成IR模型时出现如下错误:

[ ERROR ]

你可能感兴趣的:(onnx,pytorch,IR,pytorch,python)