Ubuntu 16.04中安裝Torch版的cuDNN

2020-06-16 17:38:12

Ubuntu 16.04中安裝Torch版的cuDNN這個標題也許不太合適，反正需要先安裝nvidia的cudnn，然後在安裝torch的cudnn（稱作直譯器還是什麼吧）

1. 去nvidia官網下載cudnn（註冊後便可下載）：

https://developer.nvidia.com/cudnn

我這裡下載的是cudnn-7.5-linux-x64-v5.0-ga.tgz。

2. 按照參考網址https://devtalk.nvidia.com/default/topic/912765/cudnn-install-error/中JulieBernauer給出的方法，直接使用如下命令：

sudo tar xvfz cudnn-7.5-linux-x64-v5.0-ga.tgz -C /usr/local

將檔案解壓到/usr/local/目錄下。

3. 設定環境變數（注意，cudnnv5.0版本lib資料夾為lib64）：

export CPATH=/usr/local/cuda/include:$CPATH
export LIBRARY_PATH=/usr/local/cuda/lib64:$LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

說明：上述命令不知道到底哪個地方的環境變數改了。。。我是ubuntu菜鳥。因而，直接在終端預設目錄下開啟.bashrc檔案：

gedit .bashrc

然後，簡單粗暴的把上面三句話貼上到該檔案末尾。。。如下：

190714說明：在終端中使用th後require 'cudnn'正常，但是使用zerobrane時，還是提示找不到cudnn庫。。。繼續網上搜尋， http://www.linuxidc.com/Linux/2016-07/133208.htm 裡面坑4說了：

a. sudo gedit /etc/ld.so.conf.d/cudnn.conf 就是新建一個conf檔案。名字隨便

b. 加入剛才的路徑/usr/local/cuda/lib64

c. 反正我還新增了/usr/local/cuda/include,這個估計不要也行。

d. 儲存後，再sudo ldconfig來更新快取。（可能會出現libcudnn.so.5不是符號連線的問題，不過無所謂了！！）

經過這四步之後，再開啟zerobrane，終於不報錯誤了。。。linux真是蛋疼。。。

4. 下載torch版本的cudnn，網址：https://github.com/soumith/cudnn.torch

並使用luarocks install進行安裝

5. 最最重要的一步：重新啟動電腦。。。今天下午在另一台電腦上裝完torch後，沒重新啟動，直接使用zerobrane，require包後，一直提示沒有。。。所以，一定要重新啟動。

6. 理論上就沒有問題了，如下圖：

Ubuntu 16.04中安裝Torch版的cuDNN

熱門文章