[CI] Update machine images (#9932)

tests/buildkite/infrastructure/worker-image-pipeline/linux-amd64-gpu-bootstrap.yml

@@ -12,15 +12,13 @@ phases:
             - |
               yum groupinstall -y "Development tools"
               yum install -y kernel-devel-$(uname -r)
+              dnf install -y kernel-modules-extra
               aws s3 cp --recursive s3://ec2-linux-nvidia-drivers/latest/ .
               chmod +x NVIDIA-Linux-x86_64*.run
-              CC=/usr/bin/gcc10-cc ./NVIDIA-Linux-x86_64*.run --silent
+              ./NVIDIA-Linux-x86_64*.run --silent
 
-              amazon-linux-extras install docker
-              systemctl --now enable docker
-              distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
-                 && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.repo \
-                  | sudo tee /etc/yum.repos.d/nvidia-container-toolkit.repo
+              curl -s -L https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo | tee /etc/yum.repos.d/nvidia-container-toolkit.repo
+              yum install -y nvidia-container-toolkit
               yum clean expire-cache
-              yum install -y nvidia-docker2
+              nvidia-ctk runtime configure --runtime=docker
               systemctl restart docker