Commit cdea271
authored
Merge pull request Kaggle#886 from Kaggle/tf2.3.1
Use TensorFlow 2.3.1 wheel2 files changed
Lines changed: 5 additions & 5 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,6 +1,6 @@ | |||
| 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.3.0 | ||
| 3 | + ARG TENSORFLOW_VERSION=2.3.1 | ||
| 4 | 4 | ||
| 5 | 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} | |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -1,7 +1,7 @@ | |||
| 1 | 1 | ARG BASE_TAG=staging | |
| 2 | 2 | ||
| 3 | - FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04 AS nvidia | ||
| 4 | - FROM gcr.io/kaggle-images/python-tensorflow-whl:2.3.0-py37 as tensorflow_whl | ||
| 3 | + FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 AS nvidia | ||
| 4 | + FROM gcr.io/kaggle-images/python-tensorflow-whl:2.3.1-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 | |
@@ -16,8 +16,8 @@ RUN sed -i 's/deb https:\/\/developer.download.nvidia.com/deb http:\/\/developer | |||
| 16 | 16 | # Ensure the cuda libraries are compatible with the custom Tensorflow wheels. | |
| 17 | 17 | # TODO(b/120050292): Use templating to keep in sync or COPY installed binaries from it. | |
| 18 | 18 | ENV CUDA_MAJOR_VERSION=10 | |
| 19 | - ENV CUDA_MINOR_VERSION=1 | ||
| 20 | - ENV CUDA_PATCH_VERSION=243 | ||
| 19 | + ENV CUDA_MINOR_VERSION=2 | ||
| 20 | + ENV CUDA_PATCH_VERSION=89 | ||
| 21 | 21 | ENV CUDA_VERSION=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION.$CUDA_PATCH_VERSION | |
| 22 | 22 | ENV CUDA_PKG_VERSION=$CUDA_MAJOR_VERSION-$CUDA_MINOR_VERSION=$CUDA_VERSION-1 | |
| 23 | 23 | LABEL com.nvidia.volumes.needed="nvidia_driver" | |
0 commit comments