Caffe學習系列(1)：安裝配置ubuntu14.04+cuda7.5+caffe+cudnn

1、版本

linux系統：Ubuntu 14.04 （64位）

顯卡：Nvidia K20c

cuda: cuda_7.5.18_linux.run

cudnn: cudnn-7.0-linux-x64-v4.0-rc

2、下載

Ubuntu 14.04下載地址：http://www.ubuntu.com/download/desktop （64bit)

cuda7.5下載地址：https://developer.nvidia.com/cuda-downloads/  ，下載對應的操做系統和版本cuda_7.5.18_linux.run，放到~根目錄下

cudnn下載址：https://developer.nvidia.com/cudnn, 須要註冊，並經過審覈才能下載，下載相應文件cudnn-7.0-linux-x64-v4.0-rc.tgz, 放到~根目錄下.審覈通常一至兩天，等不及的同窗可到http://pan.baidu.com/s/1bnOKBO 下載

3、開始安裝

 一、安裝ubuntu，此文不關注。

 二、禁用nouveau驅動

    按Ctrl+Alt+F1進入命令提示符,新建一個黑名單文件

# sudo vi /etc/modprobe.d/blacklist-nouveau.conf

輸入

blacklist nouveau
options nouveau modset=0

保存退出（:wq)

而後執行

# sudo update-initramfs -u

執行 lspci | grep nouveau查看是否有內容

# lspci | grep nouveau

若是沒有內容 ，說明禁用成功，若是有內容，就重啓一下再查看

# sudo reboot

重啓後，進入登陸界面的時候，不要登陸進入桌面，直接按Ctrl+Alt+F1進入命令提示符。

三、安裝cuda 7.5

先安裝一些經常使用的軟件，注意整個過程須要聯網

# sudo service lightdm stop
# sudo apt-get install g++
# sudo apt-get install git
# sudo apt-get install freeglut3-dev

接下來進入~根據目錄安裝cuda 7.5

# cd
# sudo sh cude_7.5.18_linux.run

安裝的時候，要讓你先看一堆文字（EULA)，咱們直接不停的按空格鍵到100%，而後輸入一堆accept,yes,yes或回車進行安裝。

安裝完成後，重啓，而後用ls查看一下，是否生成了四個左右以nvidia開頭的文件夾

# ls /dev/nvidia*

若是有，說明安裝成功了，若是沒有，可能不成功，須要卸載重裝。卸載命令以下：

# sudo /usr/local/cuda-7.5/bin/uninstall_cuda_7.5.pl
# sudo /usr/bin/nvidia-uninstall

若是你還不放心是否安裝成功，請參考其它教程，編譯Samples進行測試。

最後，配置環境變量，咱們直接放在系統配置文件profile裏面，先打開profile文件

# sudo vi /etc/profile

在最後面加入兩行代碼，若是你還不會用vi進行編輯，請百度

export PATH=/usr/local/cuda-7.5/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH

保存退出，至此cuda 7.5安裝完畢。

四、安裝caffe

   先下載caffe

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

而後安裝一堆第三方庫

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

接着，安裝opencv

# cd caffe
# sudo git clone https://github.com/jayrambhia/Install-OpenCV
# cd Install-OpenCV/Ubuntu
# sudo sh dependencies.sh
# cd 2.4
# sudo sh opencv2_4_10.sh

接下來，編譯caffe

# cd ~/caffe
# sudo cp Makefile.config.example Makefile.config
# make all

至此，caffe安裝完成。

五、配置運行環境

caffe運行時須要調用cuda的庫，咱們在/etc/ld.so.conf.d目錄下新建一個cafe.conf文件，將所須要用的庫的目錄寫入

# sudo vi /etc/ld.so.conf.d/caffe.conf

添加內容：

/usr/local/cuda/lib64

保存退出（：wq)

更新配置

# sudo ldconfig

六、測試caffe

  下載mnist數據

# cd ~/caffe
# sudo sh data/mnist/get_mnist.sh
# sudo sh examples/mnist/create_mnist.sh

運行時，若是你有GPU，則不須要修改配置文件，若是沒有gpu，則須要修改配置文件lenet_solver.prototxt

# sudo vi examples/mnist/lenet_solver.prototxt

將最後一行的solver_mode:GPU改成solver_mode:CPU

配置好後，就能夠運行了

# sudo sh examples/mnist/train_lenet.sh

注意，運行caffe程序時，必須在caffe的根目錄下，否則會出錯

七、安裝cudnn進行加速

假設事先已經下載好cudnn-7.0-linux-x64-v4.0-rc.tgz放到~根目錄

# cd
# sudo tar xvf cudnn-7.0-linux-x64-v4.0-rc.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.4.0.4
# sudo ln -sf libcudnn.so.4.0.4 libcudnn.so.4
# sudo ln -sf libcudnn.so.4 libcudnn.so
# sudo ldconfig

將caffek根目錄下 Makefile.config中 USE_CUDNN 行的註釋去除，而後從新進行編譯。

# cd ~/caffe
# sudo vi Makefile.config

 將USE_CUDNN 行的註釋符號#去除，即 USE_CUDNN := 1

而後從新編譯

# sudo make clean
# sudo make all

到此，所有安裝完成。

若是沒有gpu,則不能用cudnn。