sudo apt-get install libatlas-base-dev
5. 安装Caffe需要的Python包网上介绍用现有的anaconda,我反正不建议,因为路径设置麻烦,很容易出错,而且自己安装很简单也挺快的。
首先需要安装pip
sudo apt-get install python-pip
再下载caffe,我把caffe放在用户目录下
cd git clone https://github.com/BVLC/caffe.git
再转到caffe的python目录,安装scipy
cd caffe/python sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose
最后安装requirement里面的包,需要root权限
sudo su for req in $(cat requirements.txt); do pip install $req; done
如果提示报错,一般是缺少必须的包引起的,直接根据提示 pip install <package-name>就行了。
安装完后退出root权限
exit
6. 编译caffe首先修改配置文件,回到caffe目录
cd ~/caffe cp Makefile.config.example Makefile.config gedit Makefile.config
这里仅需修改两处:
i) 使用cuDNN
# USE_CUDNN := 1
这里去掉#,取消注释为
USE_CUDNN := 1
ii) 修改python包目录,这句话
PYTHON_INCLUDE := /usr/include/python2.7 \ /usr/lib/python2.7/dist-packages/numpy/core/include
改为
PYTHON_INCLUDE := /usr/include/python2.7 \ /usr/local/lib/python2.7/dist-packages/numpy/core/include
因为新安装的python包目录在这里: /usr/local/lib/python2.7/dist-packages/
接下来就好办了,直接make
make all -j4 make test make runtest make pycaffe
这时候cd 到caffe 下的 python 目录,试试caffe 的 python wrapper安装好没有:
python import caffe
如果不报错,那就说明安装好了。