From aa64f173ec4516ec80673594424bb6a1cac2447e Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Mon, 17 Dec 2018 17:49:30 +0000 Subject: [PATCH 1/2] Update to cuda 9.2 & tensorflow 1.12 --- Dockerfile | 2 +- gpu.Dockerfile | 18 +++++++++--------- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/Dockerfile b/Dockerfile index 4bec6afa..8938bf20 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,4 @@ -FROM gcr.io/kaggle-images/python-tensorflow-whl:1.11.0-py36 as tensorflow_whl +FROM gcr.io/kaggle-images/python-tensorflow-whl:1.12.0-py36 as tensorflow_whl FROM continuumio/anaconda3:5.2.0 ADD clean-layer.sh /tmp/clean-layer.sh diff --git a/gpu.Dockerfile b/gpu.Dockerfile index 38694569..1244404d 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -1,5 +1,5 @@ -FROM nvidia/cuda:9.1-cudnn7-devel-ubuntu16.04 AS nvidia -FROM gcr.io/kaggle-images/python-tensorflow-whl:1.11.0-py36 as tensorflow_whl +FROM nvidia/cuda:9.2-cudnn7-devel-ubuntu16.04 AS nvidia +FROM gcr.io/kaggle-images/python-tensorflow-whl:1.12.0-py36 as tensorflow_whl FROM gcr.io/kaggle-images/python:staging ADD clean-layer.sh /tmp/clean-layer.sh @@ -11,8 +11,8 @@ COPY --from=nvidia /etc/apt/trusted.gpg /etc/apt/trusted.gpg.d/cuda.gpg # Ensure the cuda libraries are compatible with the custom Tensorflow wheels. # TODO(b/120050292): Use templating to keep in sync or COPY installed binaries from it. -ENV CUDA_VERSION=9.1.85 -ENV CUDA_PKG_VERSION=9-1=$CUDA_VERSION-1 +ENV CUDA_VERSION=9.2.148 +ENV CUDA_PKG_VERSION=9-2=$CUDA_VERSION-1 LABEL com.nvidia.volumes.needed="nvidia_driver" LABEL com.nvidia.cuda.version="${CUDA_VERSION}" ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} @@ -33,11 +33,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ cuda-nvml-dev-$CUDA_PKG_VERSION \ cuda-minimal-build-$CUDA_PKG_VERSION \ cuda-command-line-tools-$CUDA_PKG_VERSION \ - libcudnn7=7.2.1.38-1+cuda9.0 \ - libcudnn7-dev=7.2.1.38-1+cuda9.0 \ - libnccl2=2.2.12-1+cuda9.1 \ - libnccl-dev=2.2.12-1+cuda9.1 && \ - ln -s /usr/local/cuda-9.1 /usr/local/cuda && \ + libcudnn7=7.4.1.5-1+cuda9.2 \ + libcudnn7-dev=7.4.1.5-1+cuda9.2 \ + libnccl2=2.3.7-1+cuda9.2 \ + libnccl-dev=2.3.7-1+cuda9.2 && \ + ln -s /usr/local/cuda-9.2 /usr/local/cuda && \ ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 && \ /tmp/clean-layer.sh From 580dec4f179d8f674be64dd10a3bc10d71bede31 Mon Sep 17 00:00:00 2001 From: Vincent Roseberry Date: Mon, 17 Dec 2018 20:20:28 +0000 Subject: [PATCH 2/2] add cudart-dev package --- gpu.Dockerfile | 1 + 1 file changed, 1 insertion(+) diff --git a/gpu.Dockerfile b/gpu.Dockerfile index 1244404d..945275c6 100644 --- a/gpu.Dockerfile +++ b/gpu.Dockerfile @@ -28,6 +28,7 @@ ENV NVIDIA_REQUIRE_CUDA="cuda>=9.2" RUN apt-get update && apt-get install -y --no-install-recommends \ cuda-cupti-$CUDA_PKG_VERSION \ cuda-cudart-$CUDA_PKG_VERSION \ + cuda-cudart-dev-$CUDA_PKG_VERSION \ cuda-libraries-$CUDA_PKG_VERSION \ cuda-libraries-dev-$CUDA_PKG_VERSION \ cuda-nvml-dev-$CUDA_PKG_VERSION \