from caffe2.proto import caffe2_pb2 ImportError: cannot import name caffe2_pb2

检查caffe2的gpu版本是否安装成功的时候，终端下运行python后输出

from caffe2.python import workspace

出现from caffe2.proto import caffe2_pb2 ImportError: cannot import name caffe2_pb2的错误。

但是我

import caffe2

没有出现问题

解决方案：在home路径下，按住ctrl+H，找到.bashrc文件，打开。

1. 在末尾fi后面第118行，写入

export PYTHONPATH=/home/zj/pytorch/build:/usr/local:$PYTHONPATH
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH

主要是看第一行，中间build后面的：类似于分号；的意思，就是

/usr/local和/home/zj/pytorch/build中python相关的包都作为系统python路径导入

from caffe2.proto import caffe2_pb2

用到的其实就是 /usr/local/lib/python2.7/dist-packages/caffe2/proto/caffe2_pb2.py

2.source .bashrc

然后重新进入python写入

from caffe2.python import workspace

不会报错。