!6361 modify Dockerfile for GPU version

Merge pull request !6361 from HW_KK/master
4 years ago · 522529235d
parent a8851f308b a04a58d373
commit 522529235d
2 changed files with 2 additions and 6 deletions
--- 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
--- 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