ImportError: No module named 'google'

Failed to include caffe_pb2, things might go wrong!
Traceback (most recent call last):
  File "test_model.py", line 5, in <module>
    import caffe
  File "/opt/share0/guowuwei/caffe_root/caffe/python/caffe/__init__.py", line 4, in <module>
    from .proto.caffe_pb2 import TRAIN, TEST
  File "/opt/share0/guowuwei/caffe_root/caffe/python/caffe/proto/caffe_pb2.py", line 7, in <module>
    from google.protobuf.internal import enum_type_wrapper
ImportError: No module named 'google'

 

sudo pip install protobuf or

conda install protobuf