onnx报错

最新推荐文章于 2025-01-01 23:16:02 发布

Dovy

最新推荐文章于 2025-01-01 23:16:02 发布

阅读量2.1k

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习 python 文章标签：深度学习

本文链接：https://blog.youkuaiyun.com/Simplify_boy/article/details/126874242

深度学习同时被 2 个专栏收录

2 篇文章

订阅专栏

python

1 篇文章

订阅专栏

这篇博客介绍了在Python环境中遇到ONNX库导入时出现的ImportError，该错误通常由于ONNX的依赖库protobuf未正确安装或版本不匹配导致。解决方案包括克隆ONNX源码，初始化子模块，设置环境变量并重新安装。这有助于确保ONNX及其依赖项正确编译和链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


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 .