Browse Source

modify Dockerfile for GPU version

tags/v1.0.0
wuweikang 5 years ago
parent
commit
a04a58d373
2 changed files with 2 additions and 6 deletions
  1. +1
    -3
      docker/mindspore-gpu/0.7.0-beta/Dockerfile
  2. +1
    -3
      docker/mindspore-gpu/runtime/Dockerfile

+ 1
- 3
docker/mindspore-gpu/0.7.0-beta/Dockerfile View File

@@ -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


+ 1
- 3
docker/mindspore-gpu/runtime/Dockerfile View File

@@ -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


Loading…
Cancel
Save