diff --git a/docker/cpu/Dockerfile b/docker/cpu/Dockerfile index 67e2e61bd57..28a3e381953 100644 --- a/docker/cpu/Dockerfile +++ b/docker/cpu/Dockerfile @@ -32,7 +32,7 @@ WORKDIR $CAFFE_ROOT ENV CLONE_TAG=1.0 RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/BVLC/caffe.git . && \ - pip install --upgrade pip && \ + pip install --upgrade "pip < 21.0" && \ cd python && for req in $(cat requirements.txt) pydot; do pip install $req; done && cd .. && \ mkdir build && cd build && \ cmake -DCPU_ONLY=1 .. && \ diff --git a/docker/gpu/Dockerfile b/docker/gpu/Dockerfile index dcdbdf326fb..3a40dd615aa 100644 --- a/docker/gpu/Dockerfile +++ b/docker/gpu/Dockerfile @@ -32,7 +32,7 @@ WORKDIR $CAFFE_ROOT ENV CLONE_TAG=1.0 RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/BVLC/caffe.git . && \ - pip install --upgrade pip && \ + pip install --upgrade "pip < 21.0" && \ cd python && for req in $(cat requirements.txt) pydot; do pip install $req; done && cd .. && \ git clone https://github.com/NVIDIA/nccl.git && cd nccl && make -j install && cd .. && rm -rf nccl && \ mkdir build && cd build && \