From 2947164eb833e7179c454e6e8f21bc1a4bfcb6e7 Mon Sep 17 00:00:00 2001
From: Dustin H <herbison@google.com>
Date: Mon, 15 Apr 2024 15:40:14 -0400
Subject: [PATCH] TPU VM tf2.16

http://b/334956594
---
 tpu/config.txt | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tpu/config.txt b/tpu/config.txt
index 6d6bc722..1e8299e4 100644
--- a/tpu/config.txt
+++ b/tpu/config.txt
@@ -1,9 +1,10 @@
 BASE_IMAGE=python:3.10
 PYTHON_WHEEL_VERSION=cp310
 PYTHON_VERSION_PATH=python3.10
+# gsutil ls gs://cloud-tpu-tpuvm-artifacts/tensorflow
 # https://cloud.google.com/tpu/docs/supported-tpu-configurations#libtpu_versions
-TENSORFLOW_VERSION=2.15.0
-TF_LIBTPU_VERSION=1.9.0
+TENSORFLOW_VERSION=2.16.1
+TF_LIBTPU_VERSION=1.10.1
 TF_LINUX_WHEEL_VERSION=manylinux_2_17_x86_64.manylinux2014_x86_64
 JAX_VERSION=0.4.23
 # Supports nightly