Build TensorFlow 2.4.0 whl (Kaggle#930)

Bump TensorFlow addons to 0.12 (supports TF 2.4). BUG=172352589
st01cs · Jan 11, 2021 · 94fbe63 · 94fbe63
1 parent 67cf125
commit 94fbe63
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git 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
+* `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
diff --git 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 && \