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
,以防万一。