注:本文主要介绍的是如何在ubuntu16.04系统下安装caffe以及可视化工具digits,至于cuda和cudnn的安装配置在我前一篇文章http://blog.csdn.net/cdwxx1234/article/details/75121562已经介绍了,此文不再重复。不多说了,我们开始吧!
一.必要依赖包安装
- sudoapt-getinstallbuild-essential
- sudoapt-getinstall--no-install-recommendslibboost-all-dev
- sudoapt-getinstalllibatlas-base-devlibgflags-devlibgoogle-glog-devliblmdb-dev
- sudoapt-getinstalllibprotobuf-devlibleveldb-devlibsnappy-devlibopencv-devlibhdf5-serial-devprotobuf-compiler
- sudoapt-getinstalllibblas-devliblapack-devlibatlas-base-devgfortranpython-numpy
二.安装pip和easy-install
- cd
- wget--no-check-certificatehttps://bootstrap.pypa.io/ez_setup.py
- sudopythonez_setup.py--insecure
- wgethttps://bootstrap.pypa.io/get-pip.py
- sudopythonget-pip.py
三.
安装
caffe
及
python
依赖
1).安装git
- sudoapt-getinstallgit
- gitclonehttps://github.com/BVLC/caffe.git
3).在caffe根目录python文件夹下安装依赖项;
- sudoapt-getinstallpython-pip
- sudoforreqin$(cat"requirements.txt");dopipinstall-ihttps://pypi.tuna.tsinghua.edu.cn/simple$req;done
四.编译caffe
打开终端
- cdcaffe
- sudocpMakefile.config.exampleMakefile.config
- sudogeditMakefile.config
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
- makeall–j16(16线程,线程越多速度越快)
- maketest–j16
- makeruntest–j16
- makepytest–j16
一般在编译make all 的时候可能会出错,后面出错的几率不大。碰到的问题和解决方法与安装目标检测ssd遇到的一样,具体参考http://blog.csdn.net/cdwxx1234/article/details/75136657。
编译成功后测试python
- cdcaffe/python
- python
- importcaffe
没有报错,编译就成功了。
五.安装digits
digits是caffe的一个可视化工具,可使我们对caffe的操作变得方便容易。
1).获取digits安装包
- sudogitclonehttps://github.com/NVIDIA/DIGITS.gitdigits
2).安装digits及依赖项
- cddigits
- sudoapt-getinstallgraphvizgunicorn
- forreqin$(catrequirements.txt);dosudopipinstall$req;done
3).打开digits
- cddigits
- ./digits-devserver
在网页上输入http://localhost:5000即可打开界面。
4).遇到的问题
在终端输入./digits-devserver后出现如下错误:
ValueError: Caffe executable not found inPATH
解决方案:
1).检查当前envvar的值
- echo$CAFFE_ROOT
- echo"exportCAFFE_ROOT=/home/username(我的账户名)/caffe/">>~/.profile
- source~/.profile
- echo$CAFFE_ROOT
- /home/username/caffe/
按上面的方法改了之后,digits顺利启动。