码迷,mamicode.com
首页 > 系统相关 > 详细

(原)Ubuntu16 中安装torch版的cudnn

时间:2016-07-14 01:51:14      阅读:373      评论:0      收藏:0      [点我收藏+]

标签:

转载请注明出处:

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

参考网址:

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

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

 

这个标题也许不太合适,反正需要先安装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

然后,简单粗暴的把上面三句话粘贴到该文件末尾。。。如下:

技术分享

4. 下载torch版本的cudnn,网址:https://github.com/soumith/cudnn.torch

并使用luarocks install进行安装

5. 最最重要的一步:重启电脑。。。今天下午在另一台电脑上装完torch后,没重启,直接使用zerobrane,require包后,一直提示没有。。。所以,一定要重启。

6. 理论上就没有问题了,如下图:

技术分享

(原)Ubuntu16 中安装torch版的cudnn

标签:

原文地址:http://www.cnblogs.com/darkknightzh/p/5668471.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!