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

1 篇文章 0 订阅
2 篇文章 0 订阅

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]
  • 10
    点赞
  • 34
    收藏
  • 打赏
    打赏
  • 14
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
©️2022 CSDN 皮肤主题:大白 设计师:CSDN官方博客 返回首页
评论 14

打赏作者

love_green

你的鼓励将是我创作的最大动力

¥2 ¥4 ¥6 ¥10 ¥20
输入1-500的整数
余额支付 (余额:-- )
扫码支付
扫码支付:¥2
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值