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相關信息見Ubuntu 專題頁面 http://www.linuxidc.com/topicnews.aspx?tid=2