安装GPU驱动的CentOS7自定义镜像安装nvidia显卡驱动

本文将介绍Centos7上安装NVIDIA显卡驱动。

本期实验为您介绍了如何使用安装了GPU驱动的自定义CentOS7.4镜像创建GPU规格实例并保证其正常使用，在实验正式开始前，请先完成如下准备工作：

• 购买Linux实例。具体操作步骤可参考购买云服务器；

• 确保您已经为您的Linux实例绑定了公网IP，若暂未绑定，可参考绑定公网IP；

• 云服务器状态为“运行中”；

预计部署时间：30分钟
级别：中级
相关产品：云服务器
受众：通用

操作步骤

禁用nouveau驱动

1. 依次执行以下命令完成nouveau驱动的禁用。

lsmod | grep nouveau //查看是否安装了nouveau
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
# 创建并编辑该文档，并在文档内添加以下内容：
blacklist nouveau
options nouveau modeset=0

sudo dracut --force //强制写入
reboot //为避免内核异常，请重启实例
lsmod | grep nouveau //验证是否成功禁用

卸载并重新安装Nvidia驱动

卸载Nvidia驱动（可选）

1. 执行命令/usr/bin/nvidia-uninstall卸载驱动；

2. 卸载成功后，执行命令nvidia-smi验证卸载是否成功；

3. 为避免实例异常，建议卸载完成后重启实例；

安装Nvidia驱动

1. 安装前验证环境是否满足安装条件，依次执行以下命令

yum clean all
yum makecache
yum update //更新yum源

yum install -y kernel-devel gcc //安装依赖环境
ls /boot | grep vmlinu
rpm -aq | grep kernel-devel //验证内核版本和源码版本是否一致，一致请执行下一步。

2. 安装Nvidia驱动

wget https://us.download.nvidia.com/tesla/510.85.02/NVIDIA-Linux-x86\_64-510.85.02.run //若未下载GPU驱动，请先完成下载
sudo chmod 777 NVIDIA-Linux-x86\_64-510.85.02.run //赋权
sh NVIDIA-Linux-x86\_64-510.85.02.run -no-x-check -no-nouveau-check -no-opengl-files
nvidia-smi //验证安装是否成功

安装完成且验证安装成功后，就可以正常使用该实例完成GPU训练了。

如果您有其他问题，欢迎您联系火山引擎技术支持服务。