Build new TensorFlow 2.4.1 whl. (#951)

- Also upgrade TensorFlow addons to 0.12.1
- Use bazelisk

http://b/179051776\

Author: rosbo

tensorflow-whl/CHANGELOG.md

@@ -21,4 +21,5 @@
 * `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
 * `2.4.0-py37-2`: TensorFlow 2.4.0 & m61 base image
-* `2.4.0-py37-3`: TensorFlow 2.4.0 & 7.0 CUDA compute capability
+* `2.4.0-py37-3`: TensorFlow 2.4.0 & 7.0 CUDA compute capability
+* `2.4.1-py37`: TensorFlow 2.4.1 & TFA 0.12.1

tensorflow-whl/Dockerfile

@@ -52,21 +52,17 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 
 RUN pip install --upgrade pip
 
-# See _TF_(MIN|MAX)_BAZEL_VERSION at https://github.com/tensorflow/tensorflow/blob/master/configure.py.
-ENV BAZEL_VERSION=3.1.0
-RUN apt-get install -y gnupg zip openjdk-8-jdk && \
-    apt-get install -y --no-install-recommends \
-      bash-completion \
-      zlib1g-dev && \
-    wget --no-verbose "https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel_${BAZEL_VERSION}-linux-x86_64.deb" && \
-    dpkg -i bazel_*.deb && \
-    rm bazel_*.deb
+# Use Bazelisk to ensure the proper bazel version is used.
+RUN cd /usr/local/src && \
+    wget --no-verbose "https://github.com/bazelbuild/bazelisk/releases/download/v1.7.4/bazelisk-linux-amd64" && \
+    mv bazelisk-linux-amd64 /usr/local/bin/bazel && \
+    chmod u+x /usr/local/bin/bazel
 
 # Fetch TensorFlow & install dependencies.
 RUN cd /usr/local/src && \
     git clone https://github.com/tensorflow/tensorflow && \
     cd tensorflow && \
-    git checkout tags/v2.4.0 && \
+    git checkout tags/v2.4.1 && \
     pip install keras_applications --no-deps && \
     pip install keras_preprocessing --no-deps
 
@@ -87,7 +83,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.12.0 && \
+    git checkout tags/v0.12.1 && \
     python ./configure.py && \
     bazel build --enable_runfiles build_pip_pkg && \
     bazel-bin/build_pip_pkg /tmp/tfa_cpu && \