使用AWS EC2进行AI训练进行显卡驱动配置

首先创建EC2实例选择P类型或者G类型

以下是EC2实例对应的N卡产品类型和系列，可以通过这个在官网上找到对应的驱动进行手动驱动安装

官方链接：官方驱动 | NVIDIA

创建完成登录远程ssh连接至服务器

连接以后安装gcc和make

sudo apt-get install gcc make -y

更新软件包缓存并获取实例软件包更新

sudo apt-get update -y

升级

linux-aws

程序包以接收最新版本

sud重启实例以加载最新内核版本

重启实例以加载最新内核版本

sudo reboot

重新连接实例

添加NVIDIA PPA

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

为当前运行的内核版本安装 gcc 编译器和内核标头软件包

sudo apt-get install -y gcc make linux-headers-$(uname -r)

禁用 NVIDIA 显卡的

nouveau

开源驱动程序以免与NVIDIA官方驱动可能不兼容

将

nouveau

添加到

/etc/modprobe.d/blacklist.conf

黑名单文件。

cat << EOF | sudo tee --append /etc/modprobe.d/blacklist.conf
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
EOF

编辑/etc/default/grub 文件并且添加下面的代码

GRUB_CMDLINE_LINUX="rdblacklist=nouveau"

重新生成 Grub 配置

sudo update-grub

更新initramfs，确保新的配置生效

sudo update-initramfs -u

重启系统以应用更改

sudo reboot

确认Nouveau已被禁用

lsmod | grep nouveau

如果没有输出，表示Nouveau已成功被禁用

安装推荐的或者特定功能的显卡驱动

安装 ubuntu-drivers-common

sudo apt update
sudo apt install ubuntu-drivers-common

检测推荐的驱动

ubuntu-drivers devices

可以使用ubuntu-drivers工具自动选择推荐的驱动程序

sudo ubuntu-drivers autoinstall

如果需要安装特定的N卡驱动可以先搜索可用的N卡包

apt search nvidia-driver

然后使用apt安装推荐合适的或者特定的版本

sudo apt install <package-name>

安装完成以后重启实例并且重新连接

sudo reboot

安装完成以后输入命令查看详细信息

nvidia-smi

以下是安装好的效果图

如需要手动安装指定版本或者官方驱动手动安装，点赞评论后续出教程