From 61044e81c8215e81c3e63598e7401c1389dfcddf Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Thu, 8 Jul 2021 21:58:47 +0000 Subject: [PATCH] Upgrade to TensorFlow 2.5 - Upgraded tensorflow-addons and tensorflow-gcs-config packages (requires to get TF 2.5 support). - Removed comment about `ktext` package. This packages hasn't been updated since 2019 and won't be in the future. --- Dockerfile | 6 +++--- gpu.Dockerfile | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index f24e6428..a2cc0c52 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,5 @@ ARG BASE_TAG=m73 -ARG TENSORFLOW_VERSION=2.4.1 +ARG TENSORFLOW_VERSION=2.5.0 FROM gcr.io/deeplearning-platform-release/base-cpu:${BASE_TAG} @@ -60,8 +60,8 @@ RUN pip install seaborn python-dateutil dask && \ /tmp/clean-layer.sh RUN pip install tensorflow==${TENSORFLOW_VERSION} && \ - pip install tensorflow-gcs-config==2.4.0 && \ - pip install tensorflow-addons==0.12.1 && \ + pip install tensorflow-gcs-config==${TENSORFLOW_VERSION} && \ + pip install tensorflow-addons==0.13.0 && \ /tmp/clean-layer.sh RUN apt-get install -y libfreetype6-dev && \ diff --git a/gpu.Dockerfile b/gpu.Dockerfile index e467089a..b9ca91de 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -83,7 +83,7 @@ RUN pip install jax==0.2.16 jaxlib==0.1.68+cuda$CUDA_MAJOR_VERSION$CUDA_MINOR_VE # Reinstall packages with a separate version for GPU support. RUN pip uninstall -y tensorflow && \ - pip install tensorflow-gpu==2.4.1 && \ + pip install tensorflow-gpu==2.5.0 && \ pip uninstall -y mxnet && \ pip install mxnet-cu$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION && \ /tmp/clean-layer.sh