Ubuntu完美安装显卡驱动、CUDA、cuDNN

文章目录

• 一. Ubuntu安装显卡驱动

• 二. Ubuntu安装CUDA

• 三. Ubuntu安装cuDNN

一. Ubuntu安装显卡驱动

Ubuntu安装显卡驱动最方便的方法：打开*附加驱动*，直接安装选项的*使用NVIDIA...*即可，一般来说安装第一个就行了。

安装好重启后终端输入：

nvidia-smi

有对应的输出则说明显卡驱动安装成功了。如果你的电脑是有核显和独显的，这个时候就能切换为核显正常进入Ubuntu系统了。

二. Ubuntu安装CUDA

上面输入nvidia-smi后会有一个版本数字，我们要安装的CUDA版本≤显卡驱动版本。即我上面的显卡驱动版本为12.2，那么我可以安装小于等于12.2版本的CUDA，如果你比较后发现可以安装11.8的版本的CUDA，我强烈推荐你按照这个版本。在不短的时间内，它是经典的不可撼动的存在，适配性很好。

打开官网：CUDA Toolkit Archive

找到CUDA Toolkit 11.8.0（如果你想安装别的版本的CUDA也是同理，选其他的即可），点击后根据你自己的电脑系统选择。博主的Ubuntu为22.04，选项供大家参考。

注意Installer Type选择runfile(local)，接下来在任一文件夹下的终端依次输入（如果你是选择安装的是其他版本的CUDA，安装命令在上图的下面官方会给出，把下面对应的命令替换掉即可）：

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run

sudo apt-get update

sudo apt-get upgrade

sudo apt install gcc

sudo sh cuda_11.8.0_520.61.05_linux.run

进入CUDA安装界面后按下面的选择操作，Driver和Kernel Objects不要勾选。移动到那个选项按Enter即可，之后Install.

接下来终端输入：

source ~/.bashrc

在打开的文件中的最后写入：

export PATH=$PATH:/usr/local/cuda/bin

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64

export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

默认路径应该是这样，建议去上面的路径看看有没有。

保存退出即可。最后在终端输入：

nvcc -V

有类似下图的输出则安装成功。

三. Ubuntu安装cuDNN

打开官网：cuDNN Archive | NVIDIA Developer，注意下面的cuDNN有对应版本的CUDA，不要安装错了。例如Download cuDNN v8.9.7 (December 5th, 2023), for CUDA 12.x就是对应的是12.x的版本的CUDA。Download cuDNN v8.9.7 (December 5th, 2023), for CUDA 11.x就是对应的是11.x的版本的CUDA。博主安装的是CUDA11.8，这里我选择cuDNN v8.9.7进行安装。这里选择Local Installer for Linux x86_64 (Tar)，注意需要注册或者登录你的NVIDIA账号才能进行下载。

下载完成后参照官方安装步骤：Installing cuDNN on Linux — NVIDIA cuDNN v9.2.1 documentation，即下图。

我这边直接放出相关命令方便大家。先把下载下来的tar文件解压，之后在终端依次输入：

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include

sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64

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

最后使用下面命令检测cuDNN是否成功安装：

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

有类似下面输出则成功安装cuDNN，那三个数字就是你的版本号。

---

至此，Ubuntu完美安装显卡驱动、CUDA、cuDNN。