update container images of tpu dra plugin to gke-release images

Author: syaganti

charts/tpu-dra-driver/install-tpu-dra-driver.sh

@@ -8,7 +8,7 @@ set -o pipefail
 : ${DRIVER_NAME:=tpu-dra-driver}
 
 # The registry, image and tag for the example driver
-: ${DRIVER_IMAGE_REGISTRY:="gcr.io/gke-release-staging"}
+: ${DRIVER_IMAGE_REGISTRY:="gcr.io/gke-release"}
 : ${DRIVER_IMAGE_NAME:="${DRIVER_NAME}"}
 : ${DRIVER_IMAGE_TAG:="master.0"}
 : ${DRIVER_IMAGE_PLATFORM:="ubuntu22.04"}

charts/tpu-dra-driver/templates/kubeletplugin.yaml

@@ -36,7 +36,7 @@ spec:
         {{- toYaml .Values.kubeletPlugin.podSecurityContext | nindent 8 }}
       initContainers:
       - name: "tpu-network-optimization"
-        image: "gcr.io/gke-release-staging/tpu_network_optimizer:tpu_network_optimizer_20241212.00_p0@sha256:ad1446302c2a7714af62959dd84fc9cd47552085cac4f03580cb4c5eb523e81c"
+        image: "gcr.io/gke-release/tpu_network_optimizer@sha256:ad1446302c2a7714af62959dd84fc9cd47552085cac4f03580cb4c5eb523e81c"
         securityContext:
           privileged: true
         command:
@@ -87,7 +87,7 @@ spec:
           mountPath: /sys
         - name: proc
           mountPath: /proc
-      - image: "gcr.io/gke-release-staging/vbar_control_agent@sha256:4bd33c9d448a446b8a1893f1caee93f0206674faeda82047491d144416c42c88"
+      - image: "gcr.io/gke-release/vbar_control_agent@sha256:4bd33c9d448a446b8a1893f1caee93f0206674faeda82047491d144416c42c88"
         name: vbar-control-agent
         env:
         - name: TPU_SKIP_MDS_QUERY