From c08b4d22d53e98d28c59ec1f9c254740220a2a5d Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Tue, 19 May 2020 16:29:54 +0000 Subject: [PATCH] Use TensorFlow 2.2.0 wheels --- Dockerfile | 4 ++-- gpu.Dockerfile | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index cb71ff7f..a04401f7 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,7 @@ ARG BASE_TAG=latest -ARG TENSORFLOW_VERSION=2.1.0 +ARG TENSORFLOW_VERSION=2.2.0 -FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37-3 as tensorflow_whl +FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37 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 933ca640..0cb1c8fe 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -1,7 +1,7 @@ ARG BASE_TAG=staging FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04 AS nvidia -FROM gcr.io/kaggle-images/python-tensorflow-whl:2.1.0-py37-2 as tensorflow_whl +FROM gcr.io/kaggle-images/python-tensorflow-whl:2.2.0-py37 as tensorflow_whl FROM gcr.io/kaggle-images/python:${BASE_TAG} ADD clean-layer.sh /tmp/clean-layer.sh