From a6146a6d2f227458823b342d661e7b72a55eac56 Mon Sep 17 00:00:00 2001 From: leonwanghui Date: Thu, 16 Apr 2020 17:34:53 +0800 Subject: [PATCH] Update mindspore-gpu:devel docker image --- docker/mindspore-gpu/devel/Dockerfile | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docker/mindspore-gpu/devel/Dockerfile b/docker/mindspore-gpu/devel/Dockerfile index 113bed5b06a..fe88bd9a2ce 100644 --- a/docker/mindspore-gpu/devel/Dockerfile +++ b/docker/mindspore-gpu/devel/Dockerfile @@ -6,7 +6,6 @@ MAINTAINER leonwanghui ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5 ENV CMAKE_ROOT_PATH /usr/local/cmake-3.14.1 ENV PATH ${CMAKE_ROOT_PATH}/bin:/usr/local/bin:$PATH -ENV LD_LIBRARY_PATH /usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH # Install base tools RUN apt update \ @@ -37,6 +36,9 @@ RUN DEBIAN_FRONTEND=noninteractive apt install -y \ automake \ flex +# Configure cuDNN (v7.6.5) +RUN ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.7.6.5 /usr/local/cuda/lib64/libcudnn.so + # Set bash RUN echo "dash dash/sh boolean false" | debconf-set-selections RUN DEBIAN_FRONTEND=noninteractive dpkg-reconfigure dash