Commit ddc0c00
authored
Build new TensorFlow 2.4.1 whl. (Kaggle#951)
- Also upgrade TensorFlow addons to 0.12.1
- Use bazelisk
http://b/179051776\1 parent 5bbe56d commit ddc0c00
2 files changed
Lines changed: 9 additions & 12 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -21,4 +21,5 @@ | |||
| 21 | 21 | * `2.3.1-py37-2`: TensorFlow 2.3.1 & TFA 0.11 with Python 3.7 | |
| 22 | 22 | * `2.4.0-py37`: TensorFlow 2.4.0 & Python 3.7 & TFA 0.12 | |
| 23 | 23 | * `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 | ||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -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