Skip to content

Commit ddc0c00

Browse files
authored
Build new TensorFlow 2.4.1 whl. (#951)
- Also upgrade TensorFlow addons to 0.12.1 - Use bazelisk http://b/179051776\
1 parent 5bbe56d commit ddc0c00

File tree

2 files changed

+9
-12
lines changed

2 files changed

+9
-12
lines changed

Diff for: tensorflow-whl/CHANGELOG.md

+2-1
Original file line numberDiff line numberDiff line change
@@ -21,4 +21,5 @@
2121
* `2.3.1-py37-2`: TensorFlow 2.3.1 & TFA 0.11 with Python 3.7
2222
* `2.4.0-py37`: TensorFlow 2.4.0 & Python 3.7 & TFA 0.12
2323
* `2.4.0-py37-2`: TensorFlow 2.4.0 & m61 base image
24-
* `2.4.0-py37-3`: TensorFlow 2.4.0 & 7.0 CUDA compute capability
24+
* `2.4.0-py37-3`: TensorFlow 2.4.0 & 7.0 CUDA compute capability
25+
* `2.4.1-py37`: TensorFlow 2.4.1 & TFA 0.12.1

Diff for: tensorflow-whl/Dockerfile

+7-11
Original file line numberDiff line numberDiff line change
@@ -52,21 +52,17 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
5252

5353
RUN pip install --upgrade pip
5454

55-
# See _TF_(MIN|MAX)_BAZEL_VERSION at https://github.com/tensorflow/tensorflow/blob/master/configure.py.
56-
ENV BAZEL_VERSION=3.1.0
57-
RUN apt-get install -y gnupg zip openjdk-8-jdk && \
58-
apt-get install -y --no-install-recommends \
59-
bash-completion \
60-
zlib1g-dev && \
61-
wget --no-verbose "https://github.com/bazelbuild/bazel/releases/download/${BAZEL_VERSION}/bazel_${BAZEL_VERSION}-linux-x86_64.deb" && \
62-
dpkg -i bazel_*.deb && \
63-
rm bazel_*.deb
55+
# Use Bazelisk to ensure the proper bazel version is used.
56+
RUN cd /usr/local/src && \
57+
wget --no-verbose "https://github.com/bazelbuild/bazelisk/releases/download/v1.7.4/bazelisk-linux-amd64" && \
58+
mv bazelisk-linux-amd64 /usr/local/bin/bazel && \
59+
chmod u+x /usr/local/bin/bazel
6460

6561
# Fetch TensorFlow & install dependencies.
6662
RUN cd /usr/local/src && \
6763
git clone https://github.com/tensorflow/tensorflow && \
6864
cd tensorflow && \
69-
git checkout tags/v2.4.0 && \
65+
git checkout tags/v2.4.1 && \
7066
pip install keras_applications --no-deps && \
7167
pip install keras_preprocessing --no-deps
7268

@@ -87,7 +83,7 @@ RUN pip install /tmp/tensorflow_cpu/tensorflow*.whl
8783
RUN cd /usr/local/src/ && \
8884
git clone https://github.com/tensorflow/addons && \
8985
cd addons && \
90-
git checkout tags/v0.12.0 && \
86+
git checkout tags/v0.12.1 && \
9187
python ./configure.py && \
9288
bazel build --enable_runfiles build_pip_pkg && \
9389
bazel-bin/build_pip_pkg /tmp/tfa_cpu && \

0 commit comments

Comments
 (0)