python匯入caffe模組的問題
阿新 • • 發佈:2018-12-31
出現如下問題:
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
模組。