From 53da6df8759d8ab81774548ec6671c2544762754 Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Thu, 7 Jan 2021 21:27:48 +0000 Subject: [PATCH] Build TensorFlow 2.4.0 whl Bump TensorFlow addons to 0.12 (supports TF 2.4). BUG=172352589 --- tensorflow-whl/CHANGELOG.md | 3 ++- tensorflow-whl/Dockerfile | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/tensorflow-whl/CHANGELOG.md b/tensorflow-whl/CHANGELOG.md index bcbe1603..37b8a3fe 100644 --- a/tensorflow-whl/CHANGELOG.md +++ b/tensorflow-whl/CHANGELOG.md @@ -18,4 +18,5 @@ * `2.2.0-py37-2`: TensorFlow 2.2.0 with Python 3.7 & TFA * `2.3.0-py37`: TensorFlow 2.3.0 with Python 3.7 * `2.3.1-py37`: TensorFlow 2.3.1 with Python 3.7 -* `2.3.1-py37-2`: TensorFlow 2.3.1 & TFA 0.11 with Python 3.7 \ No newline at end of file +* `2.3.1-py37-2`: TensorFlow 2.3.1 & TFA 0.11 with Python 3.7 +* `2.4.0-py37`: TensorFlow 2.4.0 & Python 3.7 & TFA 0.12 \ No newline at end of file diff --git a/tensorflow-whl/Dockerfile b/tensorflow-whl/Dockerfile index ae56e534..d9732af5 100644 --- a/tensorflow-whl/Dockerfile +++ b/tensorflow-whl/Dockerfile @@ -66,7 +66,7 @@ RUN apt-get install -y gnupg zip openjdk-8-jdk && \ RUN cd /usr/local/src && \ git clone https://github.com/tensorflow/tensorflow && \ cd tensorflow && \ - git checkout tags/v2.3.1 && \ + git checkout tags/v2.4.0 && \ pip install keras_applications --no-deps && \ pip install keras_preprocessing --no-deps @@ -87,7 +87,7 @@ RUN pip install /tmp/tensorflow_cpu/tensorflow*.whl RUN cd /usr/local/src/ && \ git clone https://github.com/tensorflow/addons && \ cd addons && \ - git checkout tags/v0.11.2 && \ + git checkout tags/v0.12.0 && \ python ./configure.py && \ bazel build --enable_runfiles build_pip_pkg && \ bazel-bin/build_pip_pkg /tmp/tfa_cpu && \