ubuntu18.04安装显卡驱动+cuda10.2+cudnn8.0.5

原创已于 2022-07-13 15:49:14 修改 · 2.9k 阅读

45 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#ubuntu #linux #python #神经网络

于 2022-07-13 15:32:28 首次发布

本文详细介绍了在Ubuntu系统中如何安装NVIDIA显卡驱动、CUDA10.2以及cuDNN8.0.5的步骤。首先通过命令行更新软件并查看推荐的显卡驱动，然后安装并验证驱动。接着，从NVIDIA官网下载并安装CUDA10.2，同时配置环境变量并验证安装。最后，下载cuDNN并将其内容复制到CUDA相关文件夹内，完成安装并验证。整个过程包括文件下载、解压、复制和修改环境变量等操作。

一、安装显卡驱动

1.更新本地软件

sudo apt-get update

2.查看系统推荐的显卡驱动

ubuntu-drivers devices

3.发现nvidia-driver-470是推荐的显卡驱动，所以输入以下命令

sudo apt-get install nvidia-driver-470

根据系统推荐灵活改变安装的版本

4.重启电脑，输入以下命令,如图所示则代表安装成功

nvidia-smi

二、安装cuda10.2，其他版本同理

1.进入cuda下载网址,找到10.2版本并如图选择信息

2.进入终端，分别输入

wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run

sudo sh cuda_10.2.89_440.33.01_linux.run

出现如下界面，选择continue

随后输入accept

最后一步，由于前面已经安装了驱动，要把安装驱动的选项取消掉，即没有叉号，如图所示，点击回车即可取消，最后选择install

3.配置环境变量

在终端中输入sudo gedit ~/.bashrc，将下列代码添加在文件底部，保存并退出。

export PATH="/usr/local/cuda-10.2/bin:$PATH" 
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH"

最后更新一下环境变量，使软连接生效

sudo source ~/.bashrc

4.验证cuda是否安装成功

nvcc -V

若如下图所示，则代表安装成功

三、安装cudnn8.0.5

1.进入cudnn官网，选择cudnn8.0.5，没有账号须注册才能下载

选择第一个进行下载

下载后，文件出现在下载里，我在下载解压的时候出现无法提取到此处，把文件名改为

cudnn-10.2-linux-x64-v8.0.5.39.tgz

将改好名字的压缩包复制到主目录下，输入以下命令进行解压，会得到一个cuda的文件夹

tar -zxvf cudnn-10.2-linux-x64-v8.0.5.39.tgz

2.复制cuDNN内容到cuda相关文件夹内

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

3.检查是否安装成功，输入以下代码，结果如图所示，则代表安装成功

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2