Update nvidia driver on host image to 530 (#13918)
- Update the nvidia driver to 530 in `image_setup.sh`
- Update the `create_templates.sh` to use the new images
- I decided to also update the CPU images just for consistency
diff --git a/TEST_FILE b/TEST_FILE
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/TEST_FILE
diff --git a/build_tools/github_actions/runner/gcp/create_image.sh b/build_tools/github_actions/runner/gcp/create_image.sh
index 0a105a3..4000b47 100755
--- a/build_tools/github_actions/runner/gcp/create_image.sh
+++ b/build_tools/github_actions/runner/gcp/create_image.sh
@@ -24,7 +24,7 @@
IMAGE_NAME="${INSTANCE_NAME/-template/}"
ZONE="${ZONE:-us-central1-a}"
PROJECT=iree-oss
-BASE_IMAGE="${BASE_IMAGE:-projects/ubuntu-os-cloud/global/images/ubuntu-2204-jammy-v20230114}"
+BASE_IMAGE="${BASE_IMAGE:-projects/ubuntu-os-cloud/global/images/ubuntu-2204-jammy-v20230531}"
GPU_MACHINE_TYPE="a2-highgpu-1g"
CPU_MACHINE_TYPE="e2-medium"
diff --git a/build_tools/github_actions/runner/gcp/create_templates.sh b/build_tools/github_actions/runner/gcp/create_templates.sh
index 8661e6a..a05d9a2 100755
--- a/build_tools/github_actions/runner/gcp/create_templates.sh
+++ b/build_tools/github_actions/runner/gcp/create_templates.sh
@@ -17,9 +17,9 @@
DRY_RUN="${DRY_RUN:-0}"
TESTING_SELF_DELETER="${TESTING_SELF_DELETER:-0}"
-GPU_IMAGE="${GPU_IMAGE:-github-runner-gpu-2023-01-30-1675109292}"
+GPU_IMAGE="${GPU_IMAGE:-github-runner-gpu-2023-06-02-1685724247}"
GPU_DISK_SIZE_GB="${GPU_DISK_SIZE_GB:-1000}"
-CPU_IMAGE="${CPU_IMAGE:-github-runner-cpu-2023-01-30-1675109033}"
+CPU_IMAGE="${CPU_IMAGE:-github-runner-cpu-2023-06-02-1685725199}"
CPU_DISK_SIZE_GB="${CPU_DISK_SIZE_GB:-1000}"
PROD_TEMPLATE_CONFIG_REPO="${PROD_TEMPLATE_CONFIG_REPO:-openxla/iree}"
diff --git a/build_tools/github_actions/runner/gcp/image_setup.sh b/build_tools/github_actions/runner/gcp/image_setup.sh
index 91494f3..8c46d45 100644
--- a/build_tools/github_actions/runner/gcp/image_setup.sh
+++ b/build_tools/github_actions/runner/gcp/image_setup.sh
@@ -276,8 +276,8 @@
# Doing these all in one command fails, probably because there's a dependency
# between them and apt-fast makes it happen in parallel. Also, it turns out
# that the Vulkan ICD is in libnvidia-gl for some reason.
- apt-get install nvidia-headless-515
- apt-get install libnvidia-gl-515-server nvidia-utils-515-server vulkan-tools
+ apt-get install nvidia-headless-530
+ apt-get install libnvidia-gl-530 nvidia-utils-530 vulkan-tools
"${script_dir}/check_cuda.sh"
"${script_dir}/check_vulkan.sh"
