最近在研究一个AI项目,需要用到英伟达显卡训练模型,centos默认使用集成显卡,没有安装Nvidia显卡驱动
yum -y install kernel-devel epel-release dkms gcc
vim /etc/default/grub
GRUB_CMDLINE_LINUX中添加rd.driver.blacklist=nouveau nouveau.modeset=0。修改后如下:
GRUB_TIMEOUT=5GRUB_DISTRIBUTOR="$(sed 's, release .*$,,g' /etc/system-release)"GRUB_DEFAULT=savedGRUB_DISABLE_SUBMENU=trueGRUB_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"
vim /etc/modprobe.d/blacklist.conf
blacklist nouveau
* 备份 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)
reboot
lsmod | grep nouveau#应该返回空
若提示lsmod命令不存在,需要安装:
yum install -y kmod
对于CentOS-7rpm -Uvh http://www.elrepo.org/elrepo-release-7.0-2.el7.elrepo.noarch.rpm对于CentOS- 6rpm -Uvh http://www.elrepo.org/elrepo-release-6-6.el6.elrepo.noarch.rpm对于CentOS- 5rpm -Uvh http://www.elrepo.org/elrepo-release-5-5.el5.elrepo.noarch.rpm
yum install -y nvidia-detectnvidia-detect
输出如下:
所以,当前Nvidia显卡型号为:kmod-nvidia-470xx
yum search kmod-nvidia-470xx
yum -y install kmod-nvidia-470xx.x86_64
因为访问nvidia网络比较慢,所以等待时间比较长
nvidia-smi
恭喜,N卡驱动已安装成功!