diff --git a/Dockerfile b/Dockerfile index a2394e16c222ecbcdaf172305a5300d022b50bfa..3affe41016a54c1c94fa8b25f79fb5f38d9bd662 100644 --- a/Dockerfile +++ b/Dockerfile @@ -25,7 +25,7 @@ COPY ./paddle/scripts/docker/root/ /root/ RUN apt-get update && \ apt-get install -y --allow-downgrades patchelf \ git python-pip python-dev python-opencv openssh-server bison \ - python3 python3-dev python3-pip \ + python3 python3-pip python3-dev \ libnccl2=2.1.2-1+cuda8.0 libnccl-dev=2.1.2-1+cuda8.0 \ wget unzip unrar tar xz-utils bzip2 gzip coreutils ntp \ curl sed grep graphviz libjpeg-dev zlib1g-dev \