- 设备版本: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版本。具体选择哪一个版本还需要根据显卡驱动来确定,二者对应关系见下表所示:
- 首先,在官网下载适用于自己设备的安装包https://developer.nvidia.com/cuda-toolkit-archive
- 命令行安装
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
在官网上下载相应的压缩包,再移动几个文件库即可,用链接文件里的一张图表示安装成功: