modify Dockerfile for GPU version

4 years ago · a04a58d373
parent 8f0c863efe
commit a04a58d373
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