fix(example): add installation of python-etcd

Signed-off-by: champon1020 <nagatelu1020@gmail.com>
champon1020 · Apr 14, 2024 · 8363165 · 8363165
1 parent c204220
commit 8363165
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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