modify Dockerfile for GPU version

This commit is contained in:
wuweikang 2020-09-16 17:30:39 +08:00
parent 8f0c863efe
commit a04a58d373
2 changed files with 2 additions and 6 deletions

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

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