Ubuntu16.04+caffe+digits安装配置

注：本文主要介绍的是如何在ubuntu16.04系统下安装caffe以及可视化工具digits，至于cuda和cudnn的安装配置在我前一篇文章http://blog.csdn.net/cdwxx1234/article/details/75121562已经介绍了，此文不再重复。不多说了，我们开始吧！

一．必要依赖包安装

sudo apt-get install build-essentialsudo apt-get install --no-install-recommends libboost-all-devsudo apt-get install libatlas-base-dev libgflags-dev libgoogle-glog-dev liblmdb-devsudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compilersudo apt-get install libblas-dev liblapack-dev libatlas-base-dev gfortran python-numpy

二．安装pip和easy-install

cdwget --no-check-certificate https://bootstrap.pypa.io/ez_setup.pysudo python ez_setup.py --insecurewget https://bootstrap.pypa.io/get-pip.pysudo python get-pip.py

三．安装caffe及python依赖

1).安装git

sudo apt-get install git

2).获取caffe源码（可cd到指定文件夹下下载）：

git clone https://github.com/BVLC/caffe.git

3).在caffe根目录python文件夹下安装依赖项；

sudo apt-get install python-pipsudo for req in $(cat "requirements.txt"); do pip install -i https://pypi.tuna.tsinghua.edu.cn/simple $req; done

四．编译caffe

打开终端

cd caffesudo cp Makefile.config.example Makefile.configsudo gedit Makefile.config

在打开的Makefile.config文件中做如下修改；

1).将USE_CUDNN := 1取消注释；

2).INCLUDE_DIRS := $(PYTHON_INCLUDE)/usr/local/include空格后然后添加/usr/include/hdf5/serial /usr/local/cuda-8.0/include/(添加cuda的include路径)；

3).LIBRARY_DIRS:=$(PYTHON_LIB)/usr/local/lib /usr/lib   空格后添加 /usr/local/cuda-8.0/lib64/(添加cuda的lib路径)；

接着就开始编译caffe

make all –j16   （16线程，线程越多速度越快）make test –j16make runtest –j16make pytest –j16

一般在编译make all 的时候可能会出错，后面出错的几率不大。碰到的问题和解决方法与安装目标检测ssd遇到的一样，具体参考http://blog.csdn.net/cdwxx1234/article/details/75136657。

编译成功后测试python 

cd caffe/pythonpython import caffe 

没有报错，编译就成功了。

五．安装digits

digits是caffe的一个可视化工具，可使我们对caffe的操作变得方便容易。

1).获取digits安装包

sudo git clone https://github.com/NVIDIA/DIGITS.git digits

2).安装digits及依赖项

cd digitssudo apt-get install graphviz gunicornfor req in $(cat requirements.txt); do sudopip install $req; done 

3).打开digits

cd digits./digits-devserver

在网页上输入http://localhost:5000即可打开界面。

4).遇到的问题

在终端输入./digits-devserver后出现如下错误：

ValueError: Caffe executable not found inPATH

解决方案：

1).检查当前envvar的值

echo $CAFFE_ROOT

2).把envvar加到~/.profile中，下次当你登录时会自动加载

echo "export CAFFE_ROOT=/home/username（我的账户名）/caffe/" >> ~/.profile

3).加载新配置

source ~/.profile

4).检查新配置

echo $CAFFE_ROOT/home/username/caffe/

按上面的方法改了之后，digits顺利启动。