1. 官方解决方案
官方解决连接 运行下面这句:
export PYTHONPATH=/path/to/caffe/python:
$PYTHONPATH
2. 方案1不管用的情况
我用的anaconda2, 用方法1会导致numpy出问题,下面这句报错。
import numpy
解决方法: 在Python脚本里添加如下:
import sys
sys.path.append(
"dir/to/caffe/python")
然后 import caffe 应该就ok了
3. 方案2的变种
方案2每次要添加2句话,我们改为1句(好像也没省什么事。。),方法如下:
vi ~/.bashrc 添加一句:
export CAFFE_ROOT=/DIR/TO/YOUR/CAFFE/python
然后 source ~/.bashrc
将下面的代码保存到 caffe_header.py
import sys, os
sys.path.append(os.getenv(
"CAFFE_ROOT") +
"/python")
将caffe_header.py 文件放到你的 PYTHONPATH 中的一个路径中(使用 echo $PYTHONPATH 查看系统里的 PYTHONPATH 有哪些,随便找一个放)。
每次 import caffe 之前,只需要加上import caffe_header 就ok 了,像下面这样
import caffe_header
import caffe
Done!
注: 每次编译caffe make -j 之后要重新编译pycaffe make pycaffe,以防万一。