歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
Linux教程網 >> Linux基礎 >> Linux教程 >> Ubuntu 16.04中安裝Torch版的cuDNN

Ubuntu 16.04中安裝Torch版的cuDNN

日期:2017/2/28 13:51:03   编辑:Linux教程

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

Copyright © Linux教程網 All Rights Reserved