From a04a58d3737f2ff83d288b6b85d3f80b5d3858a9 Mon Sep 17 00:00:00 2001 From: wuweikang Date: Wed, 16 Sep 2020 17:30:39 +0800 Subject: [PATCH] modify Dockerfile for GPU version --- docker/mindspore-gpu/0.7.0-beta/Dockerfile | 4 +--- docker/mindspore-gpu/runtime/Dockerfile | 4 +--- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/docker/mindspore-gpu/0.7.0-beta/Dockerfile b/docker/mindspore-gpu/0.7.0-beta/Dockerfile index 839995633e..ed32a8e6ed 100644 --- a/docker/mindspore-gpu/0.7.0-beta/Dockerfile +++ b/docker/mindspore-gpu/0.7.0-beta/Dockerfile @@ -35,9 +35,7 @@ RUN DEBIAN_FRONTEND=noninteractive apt install -y \ autoconf \ libtool \ automake \ - flex \ - libnccl2=2.4.8-1+cuda10.1 \ - libnccl-dev=2.4.8-1+cuda10.1 + flex # Set bash RUN echo "dash dash/sh boolean false" | debconf-set-selections diff --git a/docker/mindspore-gpu/runtime/Dockerfile b/docker/mindspore-gpu/runtime/Dockerfile index 64bf5fdb51..95c9d2a623 100644 --- a/docker/mindspore-gpu/runtime/Dockerfile +++ b/docker/mindspore-gpu/runtime/Dockerfile @@ -35,9 +35,7 @@ RUN DEBIAN_FRONTEND=noninteractive apt install -y \ autoconf \ libtool \ automake \ - flex \ - libnccl2=2.4.8-1+cuda10.1 \ - libnccl-dev=2.4.8-1+cuda10.1 + flex # Set bash RUN echo "dash dash/sh boolean false" | debconf-set-selections