Upgrade to tf2.12

http://b/281861396

Author: djherbis

Dockerfile.tmpl

@@ -36,7 +36,7 @@ RUN ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/lib
 {{ end }}
 
 # Keep these variables in sync if base image is updated.
-ENV TENSORFLOW_VERSION=2.11.0
+ENV TENSORFLOW_VERSION=2.12.0
 
 # We need to redefine the ARG here to get the ARG value defined above the FROM instruction.
 # See: https://docs.docker.com/engine/reference/builder/#understand-how-arg-and-from-interact

config.txt

@@ -1,7 +1,7 @@
 BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
 BASE_IMAGE_TAG=m108
-CPU_BASE_IMAGE_NAME=tf2-cpu.2-11.py310
-GPU_BASE_IMAGE_NAME=tf2-gpu.2-11.py310
+CPU_BASE_IMAGE_NAME=tf-cpu.2-12.py310
+GPU_BASE_IMAGE_NAME=tf-gpu.2-12.py310
 LIGHTGBM_VERSION=3.3.2
 TORCH_VERSION=2.0.0
 TORCHAUDIO_VERSION=2.0.1