From fb23bdf1e6ecb79693786bdde5bb14ebd54cd4f2 Mon Sep 17 00:00:00 2001 From: Dustin H Date: Fri, 14 Jun 2024 15:49:37 -0400 Subject: [PATCH] Upgrade to tf2.16 http://b/331651883 --- config.txt | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/config.txt b/config.txt index 6afee191..ee8193ba 100644 --- 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