这篇博客介绍了在Python环境中遇到ONNX库导入时出现的ImportError，该错误通常由于ONNX的依赖库protobuf未正确安装或版本不匹配导致。解决方案包括克隆ONNX源码，初始化子模块，设置环境变量并重新安装。这有助于确保ONNX及其依赖项正确编译和链接。
import onnx
Traceback (most recent call last):
File "torch2onnx.py", line 3, in <module>
import onnx
File "/home/dy/anaconda3/envs/torch/lib/python3.6/site-packages/onnx/__init__.py", line 5, in <module>
from .onnx_cpp2py_export import ONNX_ML
ImportError: /home/dy/anaconda3/envs/torch/lib/python3.6/site-packages/onnx/onnx_cpp2py_export.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN6google8protobuf7Message17CopyWithSizeCheckEPS1_RKS1_
解决方法：
git clone https://github.com/onnx/onnx.git
cd onnx
git submodule update --init --recursive
# Optional: prefer lite proto
export CMAKE_ARGS="-DONNX_USE_PROTOBUF_SHARED_LIBS=ON"
export CMAKE_ARGS=-DONNX_USE_LITE_PROTO=ON
pip install -e .
扫一扫
1万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
