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