TPU VM tf2.16 (#1384)

http://b/334956594
Kaggle · Apr 16, 2024 · 878dc6a · 878dc6a
1 parent e91af4d
commit 878dc6a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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.26
 # gsutil ls gs://pytorch-xla-releases/wheels/tpuvm/* | grep libtpu | grep -v -E ".*rc[0-9].*"