From f3917c87651e8c6436841241047fbb0fa1fb126b Mon Sep 17 00:00:00 2001 From: Dustin H Date: Mon, 22 Apr 2024 10:22:27 -0400 Subject: [PATCH 1/2] Revert "Keras 3 w/ tf2.16 (#1385)" This reverts commit e3181fe438b2e7dfb7b4b32cb9be71acacc9462b. --- tpu/Dockerfile | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tpu/Dockerfile b/tpu/Dockerfile index fe79260f..b11bb59d 100644 --- a/tpu/Dockerfile +++ b/tpu/Dockerfile @@ -65,10 +65,14 @@ RUN pip install tensorflow_hub https://storage.googleapis.com/cloud-tpu-tpuvm-ar papermill jupyterlab python-lsp-server[all] "jupyter-lsp==1.5.1" \ pandas matplotlib opencv-python-headless librosa accelerate diffusers scikit-learn transformers \ seaborn timm albumentations einops pyarrow fastparquet opencv-python \ - "keras>3" keras-cv keras-nlp \ kagglehub && \ /tmp/clean-layer.sh +# b/318672157 Install Keras 3 for loading models. +# We ignore dependencies for now since tf2.15 should work despite pip saying it won't. +RUN pip install --no-dependencies "keras>3" keras-cv keras-nlp && \ + /tmp/clean-layer.sh + # Tensorflow libtpu: RUN curl --output /lib/libtpu.so https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/libtpu/${TF_LIBTPU_VERSION}/libtpu.so From b4a9b56f0987e72fe0033c41898533bbc9ac15af Mon Sep 17 00:00:00 2001 From: Dustin H Date: Mon, 22 Apr 2024 10:23:29 -0400 Subject: [PATCH 2/2] Revert tf2.16 --- tpu/config.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tpu/config.txt b/tpu/config.txt index 1f21961c..c893518f 100644 --- a/tpu/config.txt +++ b/tpu/config.txt @@ -3,8 +3,8 @@ 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.16.1 -TF_LIBTPU_VERSION=1.10.1 +TENSORFLOW_VERSION=2.15.0 +TF_LIBTPU_VERSION=1.9.0 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].*"