use apt to install driver

Author: yongwww

.github/workflows/packer_build.yml

@@ -56,5 +56,4 @@ jobs:
         env:
           PKR_VAR_source_image_family: jenkins-stock-agent
           PKR_VAR_image_prefix: jenkins-gpu-agent
-          PKR_VAR_nvidia_driver_version: "570.124.06"
-          PKR_VAR_nvidia_driver_base_url: "https://us.download.nvidia.com/tesla"
+          PKR_VAR_nvidia_driver_version: "570"

packer/jenkins-agents/gpu/gpu-jenkins-agent.pkr.hcl

@@ -51,8 +51,7 @@ build {
 
   provisioner "shell" {
     environment_vars = [
-      "NVIDIA_DRIVER_VERSION=${var.nvidia_driver_version}",
-      "BASE_URL=${var.nvidia_driver_base_url}"
+      "NVIDIA_DRIVER_VERSION=${var.nvidia_driver_version}"
     ]
     execute_command = "echo 'ubuntu' | {{.Vars}} sudo -S -E bash '{{.Path}}'"
     scripts = ["${path.root}/../../scripts/nvidia-drivers.sh", "${path.root}/vulkan-setup.sh"]

packer/jenkins-agents/gpu/variables.pkr.hcl

@@ -14,6 +14,3 @@ variable "source_image_family" {
 
 variable "nvidia_driver_version" {
 }
-
-variable "nvidia_driver_base_url"{
-}

packer/scripts/nvidia-drivers.sh

@@ -1,8 +1,8 @@
 set -eux
-# install Nvidia drivers and dependencies
+
+# Install NVIDIA driver via apt (pre-built kernel modules, compatible with HWE kernels)
 apt install -y linux-headers-$(uname -r)
-curl -fSsl -O $BASE_URL/$NVIDIA_DRIVER_VERSION/NVIDIA-Linux-x86_64-$NVIDIA_DRIVER_VERSION.run
-sh NVIDIA-Linux-x86_64-$NVIDIA_DRIVER_VERSION.run -s
+apt install -y nvidia-driver-${NVIDIA_DRIVER_VERSION}-server
 
 # install nvidia-container-toolkit (replaces deprecated nvidia-container-runtime)
 curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | \