From 426347ed836e69ce80b79947c133e44f3aa13aeb Mon Sep 17 00:00:00 2001
From: Dustin Herbison <herbison@google.com>
Date: Thu, 21 Sep 2023 19:20:59 +0000
Subject: [PATCH] Upgrade to Tensorflow 2.13

http://b/301473408
---
 Dockerfile.tmpl | 7 +++----
 config.txt      | 6 +++---
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index e9a75d3e..e4ca2521 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -36,9 +36,9 @@ 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.12.0
+ENV TENSORFLOW_VERSION=2.13.0
 # See https://github.com/tensorflow/io#tensorflow-version-compatibility
-ENV TENSORFLOW_IO_VERSION=0.32.0
+ENV TENSORFLOW_IO_VERSION=0.34.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
@@ -520,8 +520,7 @@ RUN pip install flashtext \
         pyarabic \
         pandasql \
         tensorflow_hub \
-        # b/300552705 Remove pin once we upgrade to TensorFlow 2.13
-        tensorflowjs==4.10 \
+        tensorflowjs \
         jieba  \
         # ggplot is broken and main repo does not merge and release https://github.com/yhat/ggpy/pull/668
         https://github.com/hbasria/ggpy/archive/0.11.5.zip \
diff --git a/config.txt b/config.txt
index db8cee11..357cd360 100644
--- a/config.txt
+++ b/config.txt
@@ -1,7 +1,7 @@
 BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
-BASE_IMAGE_TAG=m109
-CPU_BASE_IMAGE_NAME=tf2-cpu.2-12.py310
-GPU_BASE_IMAGE_NAME=tf2-gpu.2-12.py310
+BASE_IMAGE_TAG=m111
+CPU_BASE_IMAGE_NAME=tf2-cpu.2-13.py310
+GPU_BASE_IMAGE_NAME=tf2-gpu.2-13.py310
 LIGHTGBM_VERSION=3.3.2
 TORCH_VERSION=2.0.0
 TORCHAUDIO_VERSION=2.0.1