caffe-gpu+Pytorch

Linux
1.安装Pytorch
2.用Anaconda安装caffe

conda install caffe-gpu

运行代码报错1：

Failed to include caffe_pb2, things might go wrong!
Traceback (most recent call last):
File "train_cls.py", line 105, in <module>
weights_dict = network.vgg16d.convert_caffe_to_torch(args.weights)
File "/home1/zqqHD/Smantic_Segmentation/AffinityNet/psa-master/network/vgg16d.py", line 104, in convert_caffe_to_torch
import caffe
File "/home/zqq/anaconda3/envs/pytorch/lib/python3.6/site-packages/caffe/__init__.py", line 4, in <module>
from .proto.caffe_pb2 import TRAIN, TEST
 File "/home/zqq/anaconda3/envs/pytorch/lib/python3.6/site-packages/caffe/proto/caffe_pb2.py", line 23, in <module>

解决：

pip install protobuf-py3

运行代码报错2：

Failed to include caffe_pb2, things might go wrong!
Traceback (most recent call last):
  File "train_cls.py", line 105, in <module>
    weights_dict = network.vgg16d.convert_caffe_to_torch(args.weights)
  File "/home1/zqqHD/Smantic_Segmentation/AffinityNet/psa-master/network/vgg16d.py", line 104, in convert_caffe_to_torch
    import caffe
  File "/home/zqq/anaconda3/envs/pytorch/lib/python3.6/site-packages/caffe/__init__.py", line 4, in <module>
    from .proto.caffe_pb2 import TRAIN, TEST
  File "/home/zqq/anaconda3/envs/pytorch/lib/python3.6/site-packages/caffe/proto/caffe_pb2.py", line 10, in <module>
    from google.protobuf import symbol_database as _symbol_database
  File "/home/zqq/anaconda3/envs/pytorch/lib/python3.6/site-packages/google/protobuf/symbol_database.py", line 184, in <module>
    _DEFAULT = SymbolDatabase(pool=descriptor_pool.Default())
AttributeError: module 'google.protobuf.descriptor_pool' has no attribute 'Default'

解决：

pip install --upgrade protobuf