From 63f71c4998683595167da8f9eb21ae83db3bf0ec Mon Sep 17 00:00:00 2001 From: Yizheng Huang Date: Sat, 10 Nov 2018 10:37:19 +0800 Subject: [PATCH 1/2] Update Dockerfile --- docker/cpu/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/cpu/Dockerfile b/docker/cpu/Dockerfile index 67e2e61b..d24a0a2d 100644 --- a/docker/cpu/Dockerfile +++ b/docker/cpu/Dockerfile @@ -31,7 +31,7 @@ WORKDIR $CAFFE_ROOT # https://github.com/docker/hub-feedback/issues/460 ENV CLONE_TAG=1.0 -RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/BVLC/caffe.git . && \ +RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/chuckcho/video-caffe.git . && \ pip install --upgrade pip && \ cd python && for req in $(cat requirements.txt) pydot; do pip install $req; done && cd .. && \ mkdir build && cd build && \ From 5b5f605320e13c2ff6623f4da85a21a4008489d7 Mon Sep 17 00:00:00 2001 From: Yizheng Huang Date: Sun, 11 Nov 2018 21:08:30 +0800 Subject: [PATCH 2/2] Update Dockerfile --- docker/gpu/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/gpu/Dockerfile b/docker/gpu/Dockerfile index dcdbdf32..af90377f 100644 --- a/docker/gpu/Dockerfile +++ b/docker/gpu/Dockerfile @@ -31,7 +31,7 @@ WORKDIR $CAFFE_ROOT # https://github.com/docker/hub-feedback/issues/460 ENV CLONE_TAG=1.0 -RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/BVLC/caffe.git . && \ +RUN git clone -b ${CLONE_TAG} --depth 1 https://github.com/chuckcho/video-caffe.git . && \ pip install --upgrade pip && \ 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 && \