1
- FROM nvidia/cuda:10.1 -cudnn7-devel-ubuntu18.04 AS nvidia
1
+ FROM nvidia/cuda:10.2 -cudnn7-devel-ubuntu18.04 AS nvidia
2
2
FROM gcr.io/deeplearning-platform-release/base-cpu:m46
3
3
4
4
# Avoid interactive configuration prompts/dialogs during apt-get.
@@ -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"
@@ -62,12 +62,6 @@ RUN apt-get install -y gnupg zip openjdk-8-jdk && \
62
62
dpkg -i bazel_*.deb && \
63
63
rm bazel_*.deb
64
64
65
- # CUDA 10.1 changed the location of the toolkit (libcublas): https://forums.developer.nvidia.com/t/cublas-for-10-1-is-missing/71015/17
66
- RUN cp /usr/local/cuda-10.2/targets/x86_64-linux/include/cublas_api.h /usr/local/cuda-10.1/targets/x86_64-linux/include/ && \
67
- cp /usr/local/cuda-10.2/lib64/libcublas.so.10.2.2.214 /usr/local/cuda-10.1/lib64/ && \
68
- ln /usr/local/cuda-10.1/lib64/libcublas.so.10.2.2.214 /usr/local/cuda-10.1/lib64/libcublas.so.10.1 && \
69
- ln -s /usr/local/cuda-10.1/lib64/libcublas.so.10.1 /usr/local/cuda-10.1/lib64/libcublas.so
70
-
71
65
# Fetch TensorFlow & install dependencies.
72
66
RUN cd /usr/local/src && \
73
67
git clone https://github.com/tensorflow/tensorflow && \
0 commit comments