Skip to content

Commit 2947164

Browse files
authored
TPU VM tf2.16
http://b/334956594
1 parent ee8b8fa commit 2947164

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tpu/config.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
BASE_IMAGE=python:3.10
22
PYTHON_WHEEL_VERSION=cp310
33
PYTHON_VERSION_PATH=python3.10
4+
# gsutil ls gs://cloud-tpu-tpuvm-artifacts/tensorflow
45
# https://cloud.google.com/tpu/docs/supported-tpu-configurations#libtpu_versions
5-
TENSORFLOW_VERSION=2.15.0
6-
TF_LIBTPU_VERSION=1.9.0
6+
TENSORFLOW_VERSION=2.16.1
7+
TF_LIBTPU_VERSION=1.10.1
78
TF_LINUX_WHEEL_VERSION=manylinux_2_17_x86_64.manylinux2014_x86_64
89
JAX_VERSION=0.4.23
910
# Supports nightly

0 commit comments

Comments
 (0)