1. 程式人生 > >python匯入caffe模組的問題

python匯入caffe模組的問題

出現如下問題:

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模組。