Nvidia driver update during runner start (#4429)

Nvidia driver updateduring runner start. Install required packages

Author: atalman

terraform-aws-github-runner/modules/runners-instances/templates/user-data.sh

@@ -23,11 +23,19 @@ ${install_config_runner}
 
 echo Checking if nvidia install required ${nvidia_driver_install}
 %{ if nvidia_driver_install ~}
-set +e
-sudo curl -fsL -o /tmp/nvidia_driver "https://s3.amazonaws.com/ossci-linux/nvidia_driver/NVIDIA-Linux-x86_64-535.54.03.run"
-sudo /bin/bash /tmp/nvidia_driver -s --no-drm
-sudo rm -fv /tmp/nvidia_driver
-set -e
+    set +e
+    DISTRIBUTION=$(. /etc/os-release;echo $ID$VERSION_ID)
+    if [ "${DISTRIBUTION}" != ubuntu20.04 ]; then
+        sudo yum groupinstall -y "Development Tools"
+        # ensure our kernel install is the same as our underlying kernel,
+        # groupinstall "Development Tools" has a habit of mismatching kernel headers
+        sudo yum install -y "kernel-devel-uname-r == $(uname -r)"
+        sudo modprobe backlight
+    fi
+    sudo curl -fsL -o /tmp/nvidia_driver "https://s3.amazonaws.com/ossci-linux/nvidia_driver/NVIDIA-Linux-x86_64-535.54.03.run"
+    sudo /bin/bash /tmp/nvidia_driver -s --no-drm
+    sudo rm -fv /tmp/nvidia_driver
+    set -e
 %{ endif ~}
 
 ${post_install}