python导入caffe模块的问题

来源:互联网 发布:java class.getname 编辑:程序博客网 时间:2024/06/17 22:33

出现如下问题:

from caffe import layers as L, params as P, to_protoImportError: 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.0CXX/LD -o python/caffe/_caffe.so python/caffe/_caffe.cpptouch python/caffe/proto/__init__.pyPROTOC (python) src/caffe/proto/caffe.proto

最后在CAFFE_ROOT/python/caffe/中生成_caffe.so才可以成功导入_caffe模块。