1
- FROM nvidia/cuda:10.1 -cudnn7-devel-ubuntu18.04 AS nvidia
2
- FROM gcr.io/deeplearning-platform-release/base-cpu:latest
1
+ FROM nvidia/cuda:10.2 -cudnn7-devel-ubuntu18.04 AS nvidia
2
+ FROM gcr.io/deeplearning-platform-release/base-cpu:m46
3
3
4
4
# Avoid interactive configuration prompts/dialogs during apt-get.
5
5
ENV DEBIAN_FRONTEND=noninteractive
@@ -18,8 +18,8 @@ RUN sed -i 's/deb https:\/\/developer.download.nvidia.com/deb http:\/\/developer
18
18
# Ensure the cuda libraries are compatible with the GPU image.
19
19
# TODO(b/120050292): Use templating to keep in sync.
20
20
ENV CUDA_MAJOR_VERSION=10
21
- ENV CUDA_MINOR_VERSION=1
22
- ENV CUDA_PATCH_VERSION=243
21
+ ENV CUDA_MINOR_VERSION=2
22
+ ENV CUDA_PATCH_VERSION=89
23
23
ENV CUDA_VERSION=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION.$CUDA_PATCH_VERSION
24
24
ENV CUDA_PKG_VERSION=$CUDA_MAJOR_VERSION-$CUDA_MINOR_VERSION=$CUDA_VERSION-1
25
25
LABEL com.nvidia.volumes.needed="nvidia_driver"
@@ -66,7 +66,7 @@ RUN apt-get install -y gnupg zip openjdk-8-jdk && \
66
66
RUN cd /usr/local/src && \
67
67
git clone https://github.com/tensorflow/tensorflow && \
68
68
cd tensorflow && \
69
- git checkout tags/v2.3.0 && \
69
+ git checkout tags/v2.3.1 && \
70
70
pip install keras_applications --no-deps && \
71
71
pip install keras_preprocessing --no-deps
72
72
0 commit comments