@@ -52,21 +52,17 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
52
52
53
53
RUN pip install --upgrade pip
54
54
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
64
60
65
61
# Fetch TensorFlow & install dependencies.
66
62
RUN cd /usr/local/src && \
67
63
git clone https://github.com/tensorflow/tensorflow && \
68
64
cd tensorflow && \
69
- git checkout tags/v2.4.0 && \
65
+ git checkout tags/v2.4.1 && \
70
66
pip install keras_applications --no-deps && \
71
67
pip install keras_preprocessing --no-deps
72
68
@@ -87,7 +83,7 @@ RUN pip install /tmp/tensorflow_cpu/tensorflow*.whl
87
83
RUN cd /usr/local/src/ && \
88
84
git clone https://github.com/tensorflow/addons && \
89
85
cd addons && \
90
- git checkout tags/v0.12.0 && \
86
+ git checkout tags/v0.12.1 && \
91
87
python ./configure.py && \
92
88
bazel build --enable_runfiles build_pip_pkg && \
93
89
bazel-bin/build_pip_pkg /tmp/tfa_cpu && \
0 commit comments