Ubuntu 1404 LTS 下 caffe 的安装
准备学习深度学习,装了个linux系统,然后装Cuda的时候各种心塞,装上Nvidia的驱动后Ubuntu就卡在logo界面,无法进入桌面,数次后放弃Cuda了,准备用CPU跑一些简单的代码就好了。好了,总结一下自己怎么安装的caffe。
看了很多教程都没有让自己一次装好,今天装累了去干别的事,明天继续装=-=,针对我的电脑,下面一个安装方法有效。
1. 安装开发依赖包
sudo apt-get install build-essential sudo apt-get install vim cmake git sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all2. 安装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>就行了。
如果报错为Unable to locate packet,就是无法找到包,sudo apt-get update
安装完后退出root权限exit3. 编译caffe
首先修改配置文件,回到caffe目录
cd ~/caffe cp Makefile.config.example Makefile.config gedit Makefile.config这里仅需修改一处:
修改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/
备注:将# USE_CUDNN := 1前得#注释去掉,表示使用cuDNN,如果不是使用GPU,可以将# CPU_ONLY := 1前得注释去掉。这里我们使用cuDNN来加速。接下来就好办了,直接make
make all -j4 make test make runtest make pycaffe这时候cd 到caffe 下的 python 目录,试试caffe 的 python wrapper安装好没有:
python import caffe 如果不报错,那就说明安装好了。