NVIDIA GPU驱动安装教程

引言

本文叙述如何在Windows和Linux操作系统上安装NVIDIA的驱动程序；

安装方式说明

按照NVIDIA 文档的叙述，可见对于驱动方式的安装存在多种方式；

How do I install the NVIDIA driver?
The recommended way is to use your package manager and install the cuda-drivers package (or equivalent).
When no packages are available, you should use an official “runfile”.

对于Linux 操作系统来说，是优先选择包管理器安装，没有合适的版本，才选择使用runfile 进行安装；其实Windows 同理，可以先选择下载特定的版本，特定的版本不匹配，则使用类似于runfile的方式安装；

Windows 操作系统安装

驱动下载方式1

访问驱动下载，选择对应的驱动，以3090为例：

点击搜索

就可以下载，可以看到我们所下载的驱动文件，版本是536.40

驱动下载方式2

那如果我们需要特定版本的驱动程序，如何下载？
访问CUDA，在这里，我们可以选择不同版本的驱动CUDA(Driver CUDA)，这里注意和运行的CUDA(CUDA runtime) 相区分；

以CUDA 11.7 为例，选择对应的选项，即可看到

即可下载cuda_11.7.0_516.01_windows.exe
注意这里：

• 11.7.0是指驱动的CUDA的版本号
• 516.0.01 即为驱动程序的版本

驱动安装

对于Windows，获取对应的安装程序exe文件以后，直接对照屏幕操作即可

Linux 操作系统安装

Linux 操作系统安装，以 Ubuntu 操作系统为例

禁用Nouveau驱动

为了安装NVIDIA 驱动，必须禁用Nouveau；
首先检查Nouveau是否被加载

lsmod | grep nouveau

如果有任何输出，则都说明nouveau被加载
根据官方文档，对于不同的分发版，其禁用方式不同：

只需要按照对应的操作，即可禁用Nouveau。

基于包管理器的驱动安装方式

按照官方文档 描述，使用Package Manager 安装具有完整的文档支持

并且可以选择不同的操作系统，进行具体化安装，以Ubuntu为例，给出了很详细的文档说明，在此不再赘述

基于runfile的驱动安装方式

访问CUDA，在这里，我们可以选择不同版本的驱动CUDA(Driver CUDA)，这里注意和运行的CUDA(CUDA runtime) 相区分；

以12.1为例，选择点击进入：
选择对应的操作系统配置：

特别注意选择使用 runfile(local) 安装，便可获得操作指令；
下载文件

wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run

运行文件

sudo sh cuda_12.1.0_530.30.02_linux.run

在安装CUDA 的菜单上进行选择，即可安装驱动；

安装驱动后的后继操作

安装完驱动后，如果要确保CUDA可用，也就是nvcc -V 命令起作用，需要完成后置动作；
参考官方文档 ，主要给出两步添加环境变量：

export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}

对于64位：

export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

对于32位：

export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

需要注意的是 ，这种方式只是临时添加环境变量，如果需要长期改变环境变量，则需要将对应的环境变量加入到.bashrc等配置文件
对于64位，在文件末尾追加：

PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

对于32位，在文件末尾追加：

PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib\
                         ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

结语

自此，NVIDIA驱动安装完成。