centos英伟达显卡驱动自动安装脚本

默认运行两次，第一次禁用核显，第二次安装驱动

#!/bin/sh
set -e

WORK_DIR=$(cd $(dirname "$0"); pwd)
cd $WORK_DIR

echo -e "\033[035m******************************\033[0m"
cp /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.bak
wget -O /etc/yum.repos.d/CentOS-Base.repo http://mirrors.aliyun.com/repo/Centos-7.repo

yum -y install gcc #kernel-devel kernel-headers
yum install "kernel-devel-uname-r == $(uname -r)"

a1=$(lspci | grep GeForce)

# Check if NVIDIA GPU is present
if lspci | grep -i NVIDIA > /dev/null; then
    echo "NVIDIA GPU is present."
    echo ${a1}
else
    echo "No NVIDIA GPU found."
    exit 0
fi

# Check if NVIDIA GPU drivers are installed
if lsmod | grep -q "^nvidia "; then
    echo "NVIDIA GPU drivers are installed."
else
    echo "NVIDIA GPU drivers are not installed."

    if lsmod | grep nouveau > /dev/null;then
        DEST_FILE=/lib/modprobe.d/dist-blacklist.conf
        cp ${DEST_FILE} /lib/modprobe.d/dist-blacklist.conf.bak
        # 查找包含特定字符串的行号
        line_number=$(grep -n "blacklist nvidiafb" "$DEST_FILE" | cut -d':' -f1)
        
        # 输出找到的行号
        echo "要替换的行号是: $line_number"
        # 判断是否找到了对应的行号
        if [ -n "$line_number" ]; then
            # 使用 sed 替换对应行的内容
            sed -i "${line_number}s/.*/\# blacklist nvidiafb/" "$DEST_FILE"
            echo "已将内容替换为: '# blacklist nvidiafb'"
        else
            echo "未找到匹配的行号."
        fi
        echo ""                          >> ${DEST_FILE}
        echo "# nouveau"                 >> ${DEST_FILE}
        echo "blacklist nouveau"         >> ${DEST_FILE}
        echo "options nouveau modeset=0" >> ${DEST_FILE}
            
        # 重建initramfs image步骤
        mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
        dracut /boot/initramfs-$(uname -r).img $(uname -r)
        # 修改运行级别为文本模式
        systemctl set-default multi-user.target
        echo "reboot"
        reboot
    fi
    # 安装 nvidia
    chmod +x NVIDIA-Linux-x86_64-525.147.05.run
    sh ./NVIDIA-Linux-x86_64-525.147.05.run
	systemctl get-default
	systemctl set-default graphical.target
fi
echo -e "\033[035m******************************\033[0m"

#echo "disable nouveau driver"
#DEST_FILE=/etc/modprobe.d/blacklist-nouveau.conf
#echo "blacklist nouveau"         >  ${DEST_FILE}
#echo "options nouveau modeset=0" >> ${DEST_FILE}


#lsmod | grep nouveau