配置 GPU 驱动（Linux）

更新时间：2024-04-25 15:03:14

本章节介绍 Linux 系统如何安装 NVIDIA 驱动和 CUDA。

前提条件

已获取管理控制台的账号和密码。
以 Ubuntu 操作系统的 GPU 云服务器为例。

操作步骤

安装 NVIDIA 驱动

访问 NVIDIA 驱动下载地址，搜索适用的驱动版本。选择当前创建的 GPU 云服务器所能使用的版本，点击下载进入下载页面，右键点击同意并开始下载，选择复制链接地址。
参考连接 Linux 云服务器，连接 GPU 云服务器。

使用步骤 1 中复制的链接地址，执行如下命令下载安装包。

wget https://cn.download.nvidia.com/tesla/515.65.01/NVIDIA-Linux-x86_64-515.65.01.run

执行如下命令安装 gcc，make。

sudo apt-get install gcc g++ make linux-headers-$(uname -r)

执行如下命令，并重启云服务器后生效。
```
sudo systemctl set-default multi-user.target
```
执行如下命令，查看 nouveau 驱动。
```
lsmod | grep nouveau
```
- 若结果为空，则无需处理。
- 若结果不为空，根据以下步骤进行处理：
  1. 编辑 blacklist.conf 文件。
    
    vim /etc/modprobe.d/blacklist.conf
  2. 在 blacklist.conf 文件最后加入以下内容，保存并退出。
    
    blacklist nouveau options nouveau modeset=0
  3. 重启云服务器。
执行如下命令，卸载系统中存在的驱动。
```
sudo apt-get remove nvidia-*
```
说明

若未安装其他驱动可忽略此步骤。
执行如下命令，安装驱动。
```
chmod +x NVIDIA-Linux-x86_64-515.65.01.run

./NVIDIA-Linux-x86_64-515.65.01.run
```
说明

命令行中，NVIDIA-Linux-x86_64-515.65.01.run为下载的驱动。
安装完成后，执行如下命令，查看是否安装成功。
```
nvidia-smi
```

说明
若未安装其他驱动可忽略此步骤。

说明
命令行中，`NVIDIA-Linux-x86_64-515.65.01.run`为下载的驱动。

安装 CUDA

访问 CUDA 下载地址，点击对应的版本，选择对应信息，复制安装命令进行下载。
参考连接 Linux 云服务器，连接 GPU 云服务器。

执行步骤 1 中的安装命令。

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run

执行如下命令配置环境变量。

编辑～/.bashrc 文件。
```
vim ～/.bashrc
```

在～/.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

执行以下命令使环境变量生效。
```
source ～/.bashrc
```

执行如下命令，查看是否安装成功。
```
nvcc -V
```