aws☞EC2-GPU实例安装显卡驱动

以下命令均在 ubuntu 用户中执行

安装 awscli 到 ubuntu 用户下

sudo apt-get update -y
sudo apt-get install python3-pip -y
pip3 install awscli --upgrade --user

更新系统内核和gcc，以及禁用开源驱动

sudo apt-get upgrade -y linux-aws
sudo reboot
sudo apt-get install -y gcc make linux-headers-$(uname -r)
cat << EOF | sudo tee --append /etc/modprobe.d/blacklist.conf
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv
EOF
sudo vi /etc/default/grub 
    GRUB_CMDLINE_LINUX="rdblacklist=nouveau"
sudo update-grub
sudo reboot

配置 aws key 拉取 G3 系列驱动

aws configure
# 根据提示输入相关数据，这里 aws key 可以用任意 iam 用户的

下载 G3 驱动

# G3实例直接从aws库里下载，其它需要自行去官网下载https://docs.aws.amazon.com/zh_cn/AWSEC2/latest/UserGuide/install-nvidia-driver.html#obtain-nvidia-GRID-driver-linux
aws s3 cp --recursive s3://ec2-linux-nvidia-drivers/latest/ .

安装驱动

sudo /bin/sh ./NVIDIA-Linux-x86_64*.run
# 可能会出现 gcc 版本检测，忽略掉版本检测，直接安装
sudo reboot

启用 GRUD 虚拟应用程序，默认启用 GRUD 虚拟工作站

# 具体步骤看文档
https://docs.aws.amazon.com/zh_cn/AWSEC2/latest/UserGuide/activate_grid.html

优化G3，开启全功率，关闭autoboost （P3 实例上的 GPU 不支持 autoboost 功能。）

sudo nvidia-persistenced
sudo nvidia-smi --auto-boost-default=0

# P2 实例：
sudo nvidia-smi -ac 2505,875
# P3 实例：
sudo nvidia-smi -ac 877,1530
# G3 实例：
sudo nvidia-smi -ac 2505,1177