Skip to content

Commit b5bf459

Browse files
committed
Upgrade to CUDA 10.2
1 parent c14b7fc commit b5bf459

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

tensorflow-whl/Dockerfile

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04 AS nvidia
1+
FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 AS nvidia
22
FROM gcr.io/deeplearning-platform-release/base-cpu:m46
33

44
# 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
1818
# Ensure the cuda libraries are compatible with the GPU image.
1919
# TODO(b/120050292): Use templating to keep in sync.
2020
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
2323
ENV CUDA_VERSION=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION.$CUDA_PATCH_VERSION
2424
ENV CUDA_PKG_VERSION=$CUDA_MAJOR_VERSION-$CUDA_MINOR_VERSION=$CUDA_VERSION-1
2525
LABEL com.nvidia.volumes.needed="nvidia_driver"
@@ -62,12 +62,6 @@ RUN apt-get install -y gnupg zip openjdk-8-jdk && \
6262
dpkg -i bazel_*.deb && \
6363
rm bazel_*.deb
6464

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-
7165
# Fetch TensorFlow & install dependencies.
7266
RUN cd /usr/local/src && \
7367
git clone https://github.com/tensorflow/tensorflow && \

0 commit comments

Comments
 (0)