Linux非root用户如何优雅的安装cuda和cudnn

发布时间：2018-05-23 16:11,

浏览次数：804

, 标签：

Linux

root

cuda

cudnn

Linux非root用户如何优雅的安装cuda和cudnn

跑深度学习模型时，总会遇到各种版本的问题。如深度学习框架与cuda的版本不符，或者管理安装的cuda的版本太低，因此作为非root用户就很难受。为自己装cuda和深度学习框架就非常有必要

* anaconda的必要性

* 非root安装cuda和cudnn

anaconda

* python有一个很好的版本管理器——anaconda，安装它就可以自行管理各种python版本

* anaconda 安装和使用

非root用户安装cuda和cudnn

cuda的root用户安装，网上的教程非常多，但是非root用户安装的教程就很难找到

1、cuda下载 https://developer.nvidia.com/cuda-downloads

.

2、在选择linux及对应的系统之后，选择 runfile(local)这个选项，然后下面给出的命令 “sudo sh

cuda_9.0.176_384.81_linux.run” 中sudo这个需要管理员账号，其实可以去掉sudo，直接运行，然后按照给出的提示同意协议

3、linux没有图形界面时，可以在浏览器复制下载链接“link”，然后再linux命令 “wget link”,下载下来filename.run

4、chmod +x filename.run 然后 ./filename.run

5、在协议中选择同意EULA(accept),不安装driver installation

(no),然后再安装cuda时选择个人用户的目录，如/home/yourname/cuda9,以及cudasamples的目录

6、cudnn的安装,从官网下载https://developer.nvidia.com/cudnn

(需要注册账号),

我下载的是cudnn-9.0-linux-x64-v7.tgz，然后用tar -xzvf cudnn-9.0-linux-x64-v7.tgz

解压，解压到了cuda文件夹。(cd cuda && make)可选命令

7、拷贝过去cudnn->cuda (cuda9是个人用户的下的目录/home/yourname/cuda9)

cp cuda/include/cudnn.h cuda9/include/

cp cuda /lib64/libcudnn* cuda9/lib64

chmod a+r cuda9/include/cudnn/h cuda9/lib64/libcudnn*

8、修改个人用户的环境变量

环境变量文件 ~/.bashrc位于home/yourname/~/.bashrc(用vi ~/.bashrc编辑)

“export PATH=$HOME/cuda9/bin:$PATH”

“export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/cuda9/lib64/”

这两条命令添加进去HOME目录就是/home/yourname

修改之后 source ~/.bashrc使环境变量生效

快速测试

import tensorflow as tf x = tf.nn.conv2d(tf.ones([1,1,10,1]),

tf.ones([1,5,1,1]), strides=[1, 1, 1, 1], padding='SAME') with tf.Session() as

sess: sess.run(x) # this should output a tensor of shape (1,1,10,1) with

[3,4,5,5,5,5,5,5,4,3]