centos yum安装英伟达显卡驱动

背景

最近在研究一个AI项目，需要用到英伟达显卡训练模型，centos默认使用集成显卡，没有安装Nvidia显卡驱动

操作：

1. 安装必要依赖包

yum -y install kernel-devel epel-release dkms gcc

2. 禁用默认显卡

vim /etc/default/grub

GRUB_CMDLINE_LINUX中添加rd.driver.blacklist=nouveau nouveau.modeset=0。修改后如下：

GRUB_TIMEOUT=5

GRUB_DISTRIBUTOR="$(sed 's, release .*$,,g' /etc/system-release)"

GRUB_DEFAULT=saved

GRUB_DISABLE_SUBMENU=true

GRUB_TERMINAL_OUTPUT="console"

GRUB_CMDLINE_LINUX="crashkernel=auto rd.lvm.lv=centos/root rd.lvm.lv=centos/swap rhgb quiet rd.driver.blacklist=nouveau nouveau.modeset=0 nouveau.modeset=0 rd.driver.blacklist=nouveau plymouth.ignore-udev"

GRUB_DISABLE_RECOVERY="true"  

3. 创建黑名单，加入nouveau

vim /etc/modprobe.d/blacklist.conf

blacklist nouveau

4. 备份并重新建立 initramfs image file :

* 备份 the initramfs file

$ sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak

* 重新建立 the initramfs file

$ sudo dracut -v /boot/initramfs-$(uname -r).img $(uname -r)

5. 重启系统

reboot

6. 验证nouveau 是否被成功屏蔽

lsmod | grep nouveau

#应该返回空

若提示lsmod命令不存在，需要安装：

yum install -y kmod

7.添加ELRepo源

对于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

对于CentOS- 5

rpm -Uvh http://www.elrepo.org/elrepo-release-5-5.el5.elrepo.noarch.rpm

8. 检测系统显卡型号

yum install -y nvidia-detect

nvidia-detect

输出如下：

所以，当前Nvidia显卡型号为：kmod-nvidia-470xx

9. 查找对应型号驱动

	
yum search kmod-nvidia-470xx

10. 安装Nvidia驱动

yum -y install kmod-nvidia-470xx.x86_64

因为访问nvidia网络比较慢，所以等待时间比较长

11.验证是否安装成功

nvidia-smi

恭喜，N卡驱动已安装成功！