(原)Ubuntu16 中安裝torch版的cudnn


轉載請注明出處:

http://www.cnblogs.com/darkknightzh/p/5668471.html

參考網址:

https://devtalk.nvidia.com/default/topic/912765/cudnn-install-error/

https://github.com/soumith/cudnn.torch

http://blog.csdn.net/hungryof/article/details/51557666

 

這個標題也許不太合適,反正需要先安裝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://blog.csdn.net/hungryof/article/details/51557666里面坑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. 理論上就沒有問題了,如下圖:


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM