Linux教程網 >> Linux基礎 >> Linux教程 >> Ubuntu 14.04 安裝CUDA7.5 + Caffe + cuDNN

Ubuntu 14.04 安裝CUDA7.5 + Caffe + cuDNN

日期：2017/2/28 13:47:32 编辑：Linux教程

在Ubuntu 14.04上安裝配置了Caffe深度學習框架，網上的很多教程和指導都已經過期，中間輾轉耗費了點時間，這裡把個人認為最簡單的方法整理如下。

1 版本

筆記本：ThinkPad W541
Ubuntu 14.04(64-bit)
雙顯卡：
- Intel(R) HD Graphics Family
- NVIDIA Quadro K2100M
CUDA 7.5
cuDNN

2 准備工作

系統原本是想安裝版本Wily，之前也一直在使用，但是無奈，電腦無法安裝該版本，就選用了14.04。

安裝一些基本依賴項：

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

安裝NVIDIA顯卡驅動，我是先卸載已有版本，再安裝，卸載方法：

sudo apt-get remove --purge nvidia*

重啟後登錄時，可能會循環出現填寫登錄密碼，不能進入系統的情況，按Ctrl+Alt+F1，登錄，安裝顯卡驅動：

$ sudo apt-get install nvidia-current

或者指定版本:

$ sudo add-apt-repository ppa:xorg-edgers/ppa
$ sudo apt-get update
$ sudo apt-get install nvidia-352

電腦是集成顯卡（NVIDIA獨顯的不需要這麼做），網友建議在安裝前禁掉一些服務，以保證成功。我在安裝的時候，只禁掉一項：

$ sudo service lightdm stop

（系統啟動後，不要登錄進入桌面，直接按Ctrl+Alt+F1，如果你玩過雲服務器，應該很熟悉，接著輸入系統用戶名和登錄密碼，輸入上述的指令，回到圖形界面的方法是按Ctrl+Alt+F7，記住這個操作，14.04版本出現輸入密碼不能進入桌面的情況比較常見，我安裝搜狗拼音都出現了……）

3 CUDA

網上很多安裝6.x，7.0版本的方法非常繁瑣，個人選用了相對簡單的安裝方法。

首先在官網上下載安裝文件（鏈接前文已經提供）：

我下載的是deb本地安裝文件，下載完成後，按照文檔提示的命令安裝：

$ sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda

下載了下方的Installation Guide for Linux，裡面有關於環境變量的設置方法：

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

其他安裝文件的配置方式，文檔裡也都有。

4 Caffe

下載caffe：

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

如果你沒安裝Git，請閱讀博客：Ubuntu Git安裝與使用。

然後編譯caffe：

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

配置OpenCV，請閱讀：OpenCV 3.1 Installation in Linux。

5 cuDNN

cuDNN是GPU加速計算深層神經網絡的庫（下載鏈接，前文已提供）。

本人的下載文件是：cudnn-7.0-linux-x64-v4.0-prod.tgz

在終端中切換到文件所在文件夾，輸入下面指令：

$ sudo tar xvf cudnn-7.0-linux-x64-v4.0-prod.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

（我安裝的是libcudnn.so.4.0.7，跟上面的例子對應就好）
然後切換到caffe根目錄下，將Makefile.config中的USE_CUDNN行前的#去掉：

$ cd ~/caffe
$ sudo vi Makefile.config

保存後重新編譯：

$ sudo make clean
$ sudo make all
$ sudo make test  
$ sudo make runtest

本人編譯的時候最後一步的時候，出現這樣的錯誤：

libcudart.so.7.5 cannot open shared object file: No such file or directory

網上有一堆介紹，怎麼設置環境變量的（我們分明已經設置過了），可以這樣檢查是否已經添加環境變量：

$ echo $PATH
$ echo $LD_LIBRARY_PATH

可以看到，環境變量已經添加好。

解決方法是這樣，將一些文件復制到/usr/local/lib文件夾下：

$ sudo cp /usr/local/cuda-7.5/lib64/libcudart.so.7.5 /usr/local/lib/libcudart.so.7.5 && sudo ldconfig
$ sudo cp /usr/local/cuda-7.5/lib64/libcublas.so.7.5 /usr/local/lib/libcublas.so.7.5 && sudo ldconfig
$ sudo cp /usr/local/cuda-7.5/lib64/libcurand.so.7.5 /usr/local/lib/libcurand.so.7.5 && sudo ldconfig

再次，嘗試sudo make runtest命令，出現如下：

至此，caffe安裝完成。

Ubuntu 14.04 安裝CUDA7.5 + Caffe + cuDNN PDF文件 Linux公社資源站下載：

------------------------------------------分割線------------------------------------------

免費下載地址在 http://linux.linuxidc.com/

用戶名與密碼都是www.linuxidc.com

具體下載目錄在 /2016年資料/11月/5日/Ubuntu 14.04 安裝CUDA7.5 + Caffe + cuDNN/

下載方法見 http://www.linuxidc.com/Linux/2013-07/87684.htm

------------------------------------------分割線------------------------------------------

Ubuntu 14.04 安裝配置CUDA http://www.linuxidc.com/Linux/2014-10/107501.htm

Ubuntu 12.04配置NVIDIA CUDA5.5實錄 http://www.linuxidc.com/Linux/2014-10/107502.htm

Ubuntu安裝Theano+CUDA http://www.linuxidc.com/Linux/2014-10/107503.htm