diff --git a/examples/pytorch/elastic/imagenet/Dockerfile b/examples/pytorch/elastic/imagenet/Dockerfile index 25ecc49577..01a7aae532 100644 --- a/examples/pytorch/elastic/imagenet/Dockerfile +++ b/examples/pytorch/elastic/imagenet/Dockerfile @@ -10,6 +10,9 @@ WORKDIR /workspace RUN apt-get -q update && apt-get -q install -y wget unzip RUN wget -q http://cs231n.stanford.edu/tiny-imagenet-200.zip && unzip -q tiny-imagenet-200.zip -d data && rm tiny-imagenet-200.zip +# install dependent library +RUN pip install --upgrade pip && pip install python-etcd + COPY examples/pytorch/elastic/imagenet/ ./examples USER root