Skip to content

Commit 6e16863

Browse files
authored
Merge pull request #859 from Kaggle/tf2.3
Use TensorFlow 2.3.0 wheels.
2 parents c379421 + bc50d25 commit 6e16863

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
# b/157908450 set to latest once numba 0.49.x fixes performance regression for datashader.
22
ARG BASE_TAG=m46
3-
ARG TENSORFLOW_VERSION=2.2.0
3+
ARG TENSORFLOW_VERSION=2.3.0
44

5-
FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37-2 as tensorflow_whl
5+
FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37 as tensorflow_whl
66
FROM gcr.io/deeplearning-platform-release/base-cpu:${BASE_TAG}
77

88
ADD clean-layer.sh /tmp/clean-layer.sh

gpu.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
ARG BASE_TAG=staging
22

33
FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04 AS nvidia
4-
FROM gcr.io/kaggle-images/python-tensorflow-whl:2.2.0-py37-2 as tensorflow_whl
4+
FROM gcr.io/kaggle-images/python-tensorflow-whl:2.3.0-py37 as tensorflow_whl
55
FROM gcr.io/kaggle-images/python:${BASE_TAG}
66

77
ADD clean-layer.sh /tmp/clean-layer.sh

0 commit comments

Comments
 (0)