centos安装nvidia显卡驱动

前言

当前我们使用的是 Nvidia 的显卡来运行GPU相关的算法模型，因此要求GPU服务器必须安装了对应的显卡，以及正确配置了显卡启动。
这里记录CentOS 以及 Ubuntu 系统如何能够快速配置显卡启动的方法。

显卡驱动的安装有两种方式：

从Nvidia 官网下载驱动包，手动进行安装
安装辅助检测工具，通过 yum 或 apt 的方式来进行安装
这里推荐使用 #2 方式进行操作。

一、CentOS

1. 添加ELRepo源

rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org
CentOS 7: rpm -Uvh http://www.elrepo.org/elrepo-release-7.0-2.el7.elrepo.noarch.rpm
CentOS 6: rpm -Uvh http://www.elrepo.org/elrepo-release-6-6.el6.elrepo.noarch.rpm

2. 安装显卡检测程序

执行该命令后，系统会查找官方匹配的显卡启动：

yum -y install nvidia-detect
kmod-nvidia.x86_64 : nvidia kernel module(s)
kmod-nvidia-340xx.x86_64 : nvidia-340xx kernel module(s)
kmod-nvidia-390xx.x86_64 : nvidia-390xx kernel module(s)

3. 安装对应的显卡驱动

yum -y install kmod-nvidia.x86_64
reboot

验证显卡驱动

4. 执行Nvidia显卡自带的检测工具，查看显卡及GPU相关的信息

watch -n 0.1 nvidia-smi

如果一切正常，系统会输出类似以下的信息：

±----------------------------------------------------------------------------+
| NVIDIA-SMI 440.44 Driver Version: 440.44 CUDA Version: 10.2 |
|-------------------------------±---------------------±---------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|=++==============|
| 0 GeForce RTX 2070 Off | 00000000:01:00.0 Off | N/A |
| 41% 27C P8 12W / 185W | 0MiB / 7981MiB | 0% Default |
±------------------------------±---------------------±---------------------+

±----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
±----------------------------------------------------------------------------+
安装完毕！