From 30450e2177d933d9f542ebde9d988e89e9c1877f Mon Sep 17 00:00:00 2001 From: wuweikang Date: Wed, 16 Sep 2020 17:26:59 +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 839995633e3..ed32a8e6edb 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 64bf5fdb51d..95c9d2a623d 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