系统信息: Ubuntu 16.04
anaconda3/envs/testcaffe/bin/python3
Anaconda3 4.5.2
仅限CPU
我按conda install -c caffe2 caffe2
安装了Caffe2
为了让python找到onnx我将PYTHONPATH设置为/home/username/WorkStuff/anaconda3/envs/testcaffe/lib/python3.6/site-packages/caffe2/python
在python中导入onnx import onnx
工作正常,但是当我尝试运行tutorial中提供的转换器脚本时,我得到了这个:
File "converter.py", line 2, in <module>
import caffe2.python.onnx.frontend
File "/home/username/WorkStuff/anaconda3/envs/testcaffe/lib/python3.6/site-
packages/caffe2/python/onnx/frontend.py", line 22, in <module>
from onnx import (defs, checker, helper, numpy_helper, mapping,
File "/home/username/WorkStuff/anaconda3/envs/testcaffe/lib/python3.6/site-
packages/caffe2/python/onnx/helper.py", line 9, in <module>
from onnx.backend.base import namedtupledict
File "/home/username/WorkStuff/anaconda3/envs/testcaffe/lib/python3.6/site-
packages/caffe2/python/onnx/backend.py", line 33, in <module>
from onnx import checker, GraphProto, TensorProto, AttributeProto,
ModelProto
ImportError: cannot import name 'checker'
checker.h位于/home/username/WorkStuff/anaconda3/envs/testcaffe/include/onnx
虽然我的testcaffe目录中没有... Proto文件
如果我只运行Python并编写import onnx.checker,则返回ModuleNotFoundError: No module named 'onnx.checker'
似乎onnx无法导入任何模块
有什么我只是想念或不理解? 谢谢你的帮助
答案 0 :(得分:0)
显然,较新版本的ONNX没有这些文件。
我测试了ONNX v1.0.0,一切都在那里。
请使用以下代码安装:conda install -c willyd onnx
这适用于linux x64,但我不确定如何修复其他系统。