fix(example): add installation of python-etcd in Pytorch example (kub…

…eflow#2064) Signed-off-by: champon1020 <nagatelu1020@gmail.com>
johnugeorge · Apr 28, 2024 · abf3da8 · abf3da8
1 parent ead31e1
commit abf3da8
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