出现如下问题:
from caffe import layers as L, params as P, to_proto
ImportError: No module named caffe
这是由于没有将caffe的python模块添加到python的引用目录中导致的,解决方法:
sudo vim ~/.bashrc
在文件最后写入如下内容:
export PYTHONPATH=~/caffe/python:$PYTHONPATH
该问题解决,再运行,出现如下问题:
from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
ImportError: No module named _caffe
这是由于caffe的python模块没有编译的原因,解决方法,重新编译pycaffe
sudo make pycaffe
出现如下信息:
LD -o .build_release/lib/libcaffe.so.1.0.0
CXX/LD -o python/caffe/_caffe.so python/caffe/_caffe.cpp
touch python/caffe/proto/__init__.py
PROTOC (python) src/caffe/proto/caffe.proto
最后在CAFFE_ROOT/python/caffe/
中生成_caffe.so
才可以成功导入_caffe
模块。