try 2.11

Author: djherbis

Dockerfile.tmpl

@@ -34,7 +34,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.9.2
+ENV TENSORFLOW_VERSION=2.11.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=m103
-CPU_BASE_IMAGE_NAME=tf2-cpu.2-9
-GPU_BASE_IMAGE_NAME=tf2-gpu.2-9
+CPU_BASE_IMAGE_NAME=tf2-cpu.2-11
+GPU_BASE_IMAGE_NAME=tf2-gpu.2-11
 LIGHTGBM_VERSION=3.3.2
 TORCH_VERSION=1.12.0
 TORCHAUDIO_VERSION=0.12.0