commit c1a56189e783632e238f2a88942e43456a6d3220
author Rayan Dasoriya <dasoriya@google.com> Wed Apr 13 01:40:04 2022 +0000
committer Rayan Dasoriya <dasoriya@google.com> Fri Apr 15 00:14:27 2022 +0000
tree 64c74b9094ccf0353330611ac87de80416895c88
parent de3e8c6dafc3305af2ee4811a0b518e41d436958

cos-customizer: update gpu image in cos-customizer

Update cos-customizer to use cos-gpu-installer container image which is
built from debian:bookworm. (http://go/cos-rev/31703). Also, updated the
cos-gpu-test to use cos-69-10895-201-0 instead of cos-69-10895-71-0 due
to cross compilation package issues.

BUG=b/225958311
TEST=./run_tests.sh
RELEASE_NOTE=None

Change-Id: Ia7e9e961a004b29ac0977375311d68b560f93e67
Reviewed-on: https://cos-review.googlesource.com/c/cos/tools/+/31724
Cloud-Build: GCB Service account <228075978874@cloudbuild.gserviceaccount.com>
Reviewed-by: Robert Kolchmeyer <rkolchmeyer@google.com>
Tested-by: Rayan Dasoriya <dasoriya@google.com>

src/cmd/cos_customizer/install_gpu.go

diff --git a/src/cmd/cos_customizer/install_gpu.go b/src/cmd/cos_customizer/install_gpu.go
index dd45b8c..d141c93 100644
--- a/src/cmd/cos_customizer/install_gpu.go
+++ b/src/cmd/cos_customizer/install_gpu.go
@@ -39,7 +39,7 @@
 
 const (
 	gpuScript          = "install_gpu.sh"
-	installerContainer = "gcr.io/cos-cloud/cos-gpu-installer:v20210715"
+	installerContainer = "gcr.io/cos-cloud/cos-gpu-installer:v20220414"
 )
 
 // TODO(b/121332360): Move most GPU functionality to cos-gpu-installer

testing/gpu_test.yaml

diff --git a/testing/gpu_test.yaml b/testing/gpu_test.yaml
index 4bf75b0..ebec116 100644
--- a/testing/gpu_test.yaml
+++ b/testing/gpu_test.yaml
@@ -41,6 +41,12 @@
                   --async --format='value(ID)' ."
                 "gcloud builds submit --config=testing/gpu_test/gpu_test.yaml\
                   --substitutions=_DRIVER_VERSION=460.32.03,_INPUT_IMAGE=cos-89-16108-403-42\
+                  --async --format='value(ID)' ."
+                "gcloud builds submit --config=testing/gpu_test/gpu_test.yaml\
+                  --substitutions=_DRIVER_VERSION=450.172.01,_INPUT_IMAGE=cos-93-16623-171-1\
+                  --async --format='value(ID)' ."
+                "gcloud builds submit --config=testing/gpu_test/gpu_test.yaml\
+                  --substitutions=_DRIVER_VERSION=470.82.01,_INPUT_IMAGE=cos-97-16919-29-2\
                   --async --format='value(ID)' .")
     build_ids=()
     exit_code=0

testing/gpu_test/gpu_test.yaml

diff --git a/testing/gpu_test/gpu_test.yaml b/testing/gpu_test/gpu_test.yaml
index 110a2e0..5142410 100644
--- a/testing/gpu_test/gpu_test.yaml
+++ b/testing/gpu_test/gpu_test.yaml
@@ -14,7 +14,7 @@
 
 substitutions:
   "_TEST": "gpu_test"
-  "_INPUT_IMAGE": "cos-69-10895-71-0"
+  "_INPUT_IMAGE": "cos-69-10895-201-0"
   "_INPUT_PROJECT": "cos-cloud"
   "_DRIVER_VERSION": ""
   "_DEPS_DIR": ""

testing/parallel_test.yaml

diff --git a/testing/parallel_test.yaml b/testing/parallel_test.yaml
index 16406c5..e656c21 100644
--- a/testing/parallel_test.yaml
+++ b/testing/parallel_test.yaml
@@ -16,7 +16,7 @@
   "_TEST_1": "smoke_test"
   "_INPUT_IMAGE_1": "cos-dev-69-10895-0-0"
   "_TEST_2": "gpu_test"
-  "_INPUT_IMAGE_2": "cos-69-10895-71-0"
+  "_INPUT_IMAGE_2": "cos-69-10895-201-0"
   "_INPUT_PROJECT": "cos-cloud"
 steps:
 - name: 'gcr.io/cloud-builders/bazel'