ONNX: export failure

问题

使用pytorch导出onnx(Open Neural Network Exchange)模型,结果使用conda安装完onnx之后,问题就出现了

ONNX: export failure: DLL load failed while importing _message: 找不到指定的程序

conda install onnx 

版本是0.16.2,其他依赖版本正常,使用conda search onnx --info查看package信息

在这里插入图片描述
查看网上解决方法,没有一个能够解决,包括不限于一下几种情况

其他解决

最后是使用pip包管理工具重新下载onnx,没有指定特定版本,下载最新版本0.17.0
在这里插入图片描述
安装完onnx再次转换时,出现protobuf版本问题
ONNX: export failure: No module named ‘google.protobuf.json_format’
此时protobuf和onnx都是pip管理的,所以考虑将protobuf改为conda安装
因为protobuf是conda安装的,依赖由conda进行管理
在这里插入图片描述
又爆出第一次出现的错误 _DLL load failed while importing message: 找不到指定的程序

之后将protobuf卸载,改用pip包安装protobuf,正常
在这里插入图片描述

第一个出现这个问题,是因为protobuf的问题,protobuf是conda安装的,不是因为onnx的问题
ONNX: export failure: DLL load failed while importing _message: 找不到指定的程序。

快速解决

  • 将protobuf和onnx都由pip安装
pip install protobuf
pip install onnx

留给你们测试使用包全部使用conda安装

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐