Ubuntu 14.04 安裝CUDA7.5 + Caffe + cuDNN

2020-06-16 17:29:40

在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