diff --git a/Dockerfile b/Dockerfile index c56fa1fb..cbc5290f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,7 +2,7 @@ ARG BASE_TAG=m46 ARG TENSORFLOW_VERSION=2.3.1 -FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37 as tensorflow_whl +FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37-2 as tensorflow_whl FROM gcr.io/deeplearning-platform-release/base-cpu:${BASE_TAG} ADD clean-layer.sh /tmp/clean-layer.sh diff --git a/gpu.Dockerfile b/gpu.Dockerfile index 35f78cd4..2c0f2556 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -1,7 +1,7 @@ ARG BASE_TAG=staging FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 AS nvidia -FROM gcr.io/kaggle-images/python-tensorflow-whl:2.3.1-py37 as tensorflow_whl +FROM gcr.io/kaggle-images/python-tensorflow-whl:2.3.1-py37-2 as tensorflow_whl FROM gcr.io/kaggle-images/python:${BASE_TAG} ADD clean-layer.sh /tmp/clean-layer.sh diff --git a/tensorflow-whl/CHANGELOG.md b/tensorflow-whl/CHANGELOG.md index ae403d24..bcbe1603 100644 --- a/tensorflow-whl/CHANGELOG.md +++ b/tensorflow-whl/CHANGELOG.md @@ -17,4 +17,5 @@ * `2.2.0-py37`: TensorFlow 2.2.0 with Python 3.7 * `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.6 \ No newline at end of file +* `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 \ No newline at end of file diff --git a/tensorflow-whl/Dockerfile b/tensorflow-whl/Dockerfile index d88604af..ae56e534 100644 --- a/tensorflow-whl/Dockerfile +++ b/tensorflow-whl/Dockerfile @@ -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.10.0 && \ + git checkout tags/v0.11.2 && \ python ./configure.py && \ bazel build --enable_runfiles build_pip_pkg && \ bazel-bin/build_pip_pkg /tmp/tfa_cpu && \