Ubuntu16.04+CUDA8.0+cudnn7.5+Caffe安装过程
1. 主要用到的软件列表:
注:具体版本由自己电脑决定,官网下载超链接。
Ubuntu-16.04.1-desktop-amd64(64位ubuntu)
cudnn-7.5-linux-x64-v5.1.tgz(需要注册才能下载,现在审核时间很短,填个问卷就能下载了)
caffe-master (也可以在线下载 )
@H_403_10@
2. cpu版Caffe
直入正题,因为先前只需要安装cpu版的caffe,所以实现起来不是很难,参照这篇博客,基本能够实现,至于他最后训练手写字体mnist数据库失败的原因是因为没有修改example/mnist文件夹中的运行配置文件,即将默认的GPU运行模式改成cpu。
@H_403_10@ 3. GPU版本安装及配置
3.1 安装一些基本依赖库
- sudoapt-getinstallbuild-essential
- sudoapt-getinstalllibprotobuf-devlibleveldb-devlibsnappy-devlibopencv-devlibboost-all-devlibhdf5-serial-devlibgflags-devlibgoogle-glog-devliblmdb-devprotobuf-compiler
3.2 更新gcc和g++
@H_403_10@
下载安装gcc/g++-5
- sudoadd-apt-repositoryppa:ubuntu-toolchain-r/test
- sudoapt-getupdate
- sudoapt-getinstallgcc-5
- sudoapt-getinstallg++-5
@H_403_10@
改变系统默认的gcc/g++编译器版本,换成gcc/g++-5
- sudosu
- cd../../usr/bin
- ln-s/usr/bin/g++-5/usr/bin/g++-f
- ln-s/usr/bin/gcc-5/usr/bin/gcc-f
这样,gcc就默认成gcc-5,g++也默认成g++-5了。
3.3 安装CUDA8.0
- sudodpkg-icuda-repo-<distro>_<version>_<architecture>.deb#自己的.deb包名
- sudoapt-getupdate
- sudoapt-getinstallcuda
3.4 安装cudnn7.5
将上面提到的cudnn安装包下载好解压,进入解压后的文件,在终端执行下面的指令安装:
- cdcuda
- sudocplib64/lib*/usr/local/cuda/lib64/
- sudocpinclude/cudnn.h/usr/local/cuda/include/
@H_403_10@
然后更新网络连接:
- cd/usr/local/cuda/lib64/
- sudochmod+rlibcudnn.so.5.1.3#自己查看.so的版本
- sudoln-sflibcudnn.so.5.1.3libcudnn.so.5
- sudoln-sflibcudnn.so.5libcudnn.so
- sudoldconfig
3.5 添加环境变量
编辑计算机文件夹下的/etc/profile,加入CUDA环境变量(下面的2,3句),保存。
- sudogedit/etc/profile
- PATH=/usr/local/cuda/bin:$PATH
- exportPATH
- source/etc/profile
- sudoldconfig
3.6 Build CUDA Sample
进入usr/local/cuda/samples,然后build samples,命令如下:
@H_403_10@
4 安装Atlas
安装命令如下:
- shsudo./opencv2_4_10.sh
@H_403_10@
6 配置Python环境
- INCLUDE_DIRS+=$(BUILD_INCLUDE_DIR)./src./include
- INCLUDE_DIRS+=$(BUILD_INCLUDE_DIR)./src./include/usr/include/hdf5/serial
- LIBRARIES+=gloggflagsprotobufboost_systemboost_filesystemmhdf5_hlhdf5
- LIBRARIES+=gloggflagsprotobufboost_systemboost_filesystemmhdf5_serial_hlhdf5_serial
@H_403_10@
保存之后,进入caffe-master/python文件,在终端执行下面的命令:
- sudoapt-getinstallpython-pippython-devbuild-essential
- sudopipinstall--upgradepip
- sudopipinstall-rpython/requirements.txt
@H_403_10@
7 安装ipython notebook
notebook是很方便的python编译和运行工具,值得拥有 ,随便在哪个终端执行下面命令。
- sudoapt-getinstallipython-notebookpython-sympy
- sudopipinstalljupyter
- mkdirnotebook
- cdnotebook
- ipythonnotebook
@H_403_10@
8 编译Caffe
- cpMakefile.config.exampleMakefile.config
@H_403_10@
然后打开Makefile.config文件,将里面的一些配置修改一下,把注释去掉,后面数字修改一下:
- USE_CUDNN:=1
- USE_OPENCV:=1
- USE_LMDB:=1
- WITH_PYTHON_LAYER:=1
- CUSTOM_CXX:=g++
@H_403_10@
这是为了运行下面mnist例子做准备的,不同的例子,配置文件修改的内容可能不一样,这也是留一个备份的原因。
@H_403_10@ 然后编译下面4个命令:
- makeall-j8
- makepycaffe-j8
- maketest-j8
- makeruntest-j8
如果没有报错,那基本上就安装成功啦!
9 mnist手写字体识别例子实现
进入caffe-master文件夹,打开终端,下载数据:- ./data/mnist/get_mnist.sh
- ./examples/mnist/create_mnist.sh
@H_403_10@
我自己在安装的过程中借鉴了很多别人的经验,觉得这就是博客的魅力所在,感谢他们的无私分享,也希望自己的分享能够给其他人带来帮助,有任何疑问都可以交流哈。
@H_403_10@
@H_403_10@
@L_403_132@10 References
1.http://blog.csdn.net/letian0805/article/details/52845285
2.http://blog.csdn.net/ubunfans/article/details/47724341