caffe 安装(Ubuntu 14.04 +caffe+GTX 1080+CUDA8.0 +cuDNN v5)

一、硬件与环境

显卡:GTX 1080
系统:Ubuntu 14.04
CUDA:cuda_8.0.44_linux.run(一定是下载这个)
cuDNN:cudnn-8.0-linux-x64-v5.1.tgz

GTX1080显卡必须用CUDA 8.0版本。切记,千万不要下载 deb 包,否则后方无数坑在等着你。GTX1080显卡必须用cuDNN-8.0-V5.1版本,不然用 caffe 跑模型,用 CPU或GPU显卡跑精度正常,一旦开启cuDNN模式,精度(acc)立刻下降到 0.1 左右,loss 非常大。下载需注册。最好注册一个账号,选择对应的版本,不要用网上其他教程给的现成的包,出问题的概率非常大。

二、安装

1 安装依赖库

sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

2 安装 BLAS

BLAS (Basic Linear Algebra Subprograms)是一个基本线性代数运算库,除了BLAS还有Intel的MKL(Math Kernel Library),这个库需要购买或者用edu邮箱申请,相应的这个库比BLAS运算效率高,如果有GPU的情况下这个库就不是很重要了,所以这里我们用免费的BLAS,安装命令如下。

sudo apt-get install libatlas-base-dev

3 安装opencv

OpenCV的全称是:Open Source Computer Vision Library。OpenCV是一个基于BSD许可(开源)发行的跨平台计算机视觉库,可以运行在Linux、Windows和Mac OS操作系统上。它轻量级而且高效——由一系列 C 函数和少量 C++ 类构成,同时提供了Python、Ruby、MATLAB等语言的接口,实现了图像处理和计算机视觉方面的很多通用算法。caffe里用到很多caffe的函数,包括基本的图像处理,需要执行以下命令:

3.1 安装cmake等编译opencv需要用到的工具

sudo apt-get install build-essential
sudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev

3.2 下载opencv

mkdir ~/opencv
cd ~/opencv
wget https://github.com/Itseez/opencv/archive/3.0.0-alpha.zip -O opencv-3.0.0-alpha.zip
unzip opencv-3.0.0-alpha.zip
cd opencv-3.0.0-alpha

3.3 编译安装opencv

mkdir build
cd build
cmake ..
sudo make
sudo make install
sudo /bin/bash -c 'echo "/usr/local/lib" > /etc/ld.so.conf.d/opencv.conf'
sudo ldconfig

4 安装CUDA

CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。 它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。 开发人员现在可以使用C语言来为CUDA™架构编写程序,C语言是应用最广泛的一种高级编程语言。caffe中几乎所有的layer都有GPU实现的版本,利用cuda使得训练速度大大提升。CUDA相应也是比较难以安装的一个依赖包了,这次安装的过程 中出现最多问题的就是CUDA的安装。

在安装之前需要说明一些情况,Ubuntu14.04安装的时候自带了一个名为nouveau的驱动,这个驱动为安装带来了很多麻烦。如果你原本的显示设备是NVIDIA的卡的话,尤其是单GPU的话,那你安装的时候需要先关闭GUI,执行Ctrl+Alt+F1后进入tty1,执行sudo stop lightdm关闭显示器管理器,禁用旧的显卡驱动:

cd /etc/modprobe.d/
sudo vim nvidia-installer-disable-nouveau.conf 

输入以下内容,保存并退出:

blacklist nouveau options nouveau modeset=0 

打开/etc/default/grub,在文件末尾添加:

rdblacklist nouveau nouveau.modeset=0 

然后再安装NVIDIA驱动:

sudo sh ./NVIDIA-Linux-x86_64-352.30.run

安装CUDA成功之后再执行

sudo start lightdm

5 安装 cuDNN

cd ~
sudo tar xvf cudnn-8.0-linux-x64-v5.1.tgz
cd cuda/include
sudo cp *.h /usr/local/include/
cd ../lib64
sudo cp lib* /usr/local/lib/
cd /usr/local/lib# sudo chmod +r libcudnn.so.5.1.5
sudo ln -sf libcudnn.so.5.1.5 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig

注意:libcudnn.so后面跟的数字可能和你下载的 cudnn 包小版本的不同而不同,去~/cuda/lib64下看一眼,相对应地进行修改。

6 安装 caffe

6.1 下载 caffe

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

6.2安装依赖

sudo apt-get install libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler

6.3 编译 caffe

cd ~/caffe
cp Makefile.config.example Makefile.config
vim Makefile.config

修改Makefile.config文件:
USE_CUDNN := 1, 开头的#删除
USE_OPENCV := 0,开头的#删除
OPENCV_VERSION := 3,开头的#删除 因为我安装的是opencv3

make all
make test
make runtest

OK现在就完成了,体验下1080显卡的霸气。

你可能感兴趣的:(caffe学习笔记)