ubuntu18.04安装cuda10.1和对应的cudnn

  • 设备版本:ubuntu18.04 RTX3070 NVIDIA Drive 455.38
    我的显卡驱动在装好系统之后已经安装好了,这里就不做介绍了,直接安装cuda和cudnn。想看显卡驱动安装方法的小伙伴可以单击这里

作者补充:
按本文方法安装455.38+cuda10.1后发现,显卡驱动455.38不支持cuda11.0以下的版本,不过现在Pytorch也支持CUDA11.0了,可以降低显卡驱动的版本按本文方法安装CUDA,也可以按本文方法直接安装CUDA11.0。

安装cuda10.1

因为深度学习目前很多研究的开源代码不兼容最新的tensorflow和pytorch,所以选择了较适用的cuda10.1或者cuda10.2版本。具体选择哪一个版本还需要根据显卡驱动来确定,二者对应关系见下表所示:
在这里插入图片描述

1. cd进入cuda安装包(.run文件)所在的目录下,打开终端
2. sudo sh cuda_10.1.105_418.39_linux.run  					//具体的文件名根据自己的改一下

我自己操作的时候没有存图,所以参考了这个链接里的两张图,特此说明:
https://blog.csdn.net/sss_369/article/details/94591280
在这里插入图片描述
因为已经安装过更高版本的显卡驱动了(455.38),所以这里移动光标到Driver,按Enter,将[x]改为[ ],再选择install进行安装。
在这里插入图片描述

  • 安装完成之后,需要添加环境变量:
sudo gedit ~/.bashrc

#在打开的文件最后添加如下代码
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1
#保存,退出

#输入下面的代码使之生效
source ~/.bashrc

确认cuda安装成功:
在这里插入图片描述

安装cudnn

这一步可以参考这篇经验贴:
https://blog.csdn.net/wanzhen4330/article/details/81699769
官网上下载相应的压缩包,再移动几个文件库即可,用链接文件里的一张图表示安装成功:
在这里插入图片描述

  • 1
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 6
    评论

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

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值