配置Makefile.config文件,需要修改如下配置至相应的路径(参照配置文件的注释,疑惑的地方google即可):
1、CUDA_DIR
2、BLAS := open,BLAS_INCLUDE :=,BLAS_LIB :
3、PYTHON_INCLUDE :=
注意此处需要添加python下numpy的路径
4、PYTHON_LIB :=
5、WITH_PYTHON_LAYER := 1
注意此处需要将python的hdf5安装路径添加到:INCLUDE_DIRS := $(PYTHON_INCLUDE)和LIBRARY_DIRS := $(PYTHON_LIB)
上述配置文件修改好后,回到caffe的根目录,执行下列语句:
make all -j4 (注意此处进程不要过多,否则会出现诡异的错误)
make test -j4
make runtest -j4
至此caffe主程序编译完毕。
下面编译pycaffe,执行
make pycaffe
执行完后修改~/.bashrc文件,添加
export PYTHONPATH=“{caffe_dir}/python:$PYTHONPATH”
使得python能够找到caffe的依赖,然后执行命令:
source ~/.bashrc
验证pycaffe是否安装正确:
进入python,import caffe,如果成功则说明一切ok