Upgrade to tf2.16

http://b/331651883
Kaggle · Jun 14, 2024 · fb23bdf · fb23bdf
1 parent e0587aa
commit fb23bdf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/config.txt b/config.txt
@@ -1,7 +1,7 @@
 BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
-BASE_IMAGE_TAG=m114
-CPU_BASE_IMAGE_NAME=tf2-cpu.2-15.py310
-GPU_BASE_IMAGE_NAME=tf2-gpu.2-15.py310
+BASE_IMAGE_TAG=m122
+CPU_BASE_IMAGE_NAME=tf2-cpu.2-16.py310
+GPU_BASE_IMAGE_NAME=tf2-gpu.2-16.py310
 LIGHTGBM_VERSION=4.2.0
 TORCH_VERSION=2.1.2
 TORCHAUDIO_VERSION=2.1.2