您的位置:首页 > 编程语言 > Python开发

python导入caffe模块的问题

2017-08-30 20:30 1091 查看
出现如下问题:

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
模块。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python caffe