Commit bc50d25
committed
Use TensorFlow 2.3.0 wheels.
PR creating the TensorFlow 2.3.0 whl: Kaggle#857.
BUG=1618052801 parent c379421 commit bc50d25
2 files changed
Lines changed: 3 additions & 3 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,8 +1,8 @@ | |||
| 1 | 1 | # b/157908450 set to latest once numba 0.49.x fixes performance regression for datashader. | |
| 2 | 2 | ARG BASE_TAG=m46 | |
| 3 | - ARG TENSORFLOW_VERSION=2.2.0 | ||
| 3 | + ARG TENSORFLOW_VERSION=2.3.0 | ||
| 4 | 4 | ||
| 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 | ||
| 6 | 6 | FROM gcr.io/deeplearning-platform-release/base-cpu:${BASE_TAG} | |
| 7 | 7 | ||
| 8 | 8 | ADD clean-layer.sh /tmp/clean-layer.sh | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,7 +1,7 @@ | |||
| 1 | 1 | ARG BASE_TAG=staging | |
| 2 | 2 | ||
| 3 | 3 | 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 | ||
| 5 | 5 | FROM gcr.io/kaggle-images/python:${BASE_TAG} | |
| 6 | 6 | ||
| 7 | 7 | ADD clean-layer.sh /tmp/clean-layer.sh | |
0 commit comments