Get the elasticdl_ps path dynamically while building the docker image for elasticdl job. (#2130)

Author: brightcoder01

elasticdl_client/api.py

@@ -53,7 +53,9 @@ def init_zoo(args):
  --extra-index-url={{ EXTRA_PYPI_INDEX }}
 
 RUN pip install elasticdl --extra-index-url={{ EXTRA_PYPI_INDEX }}
-ENV PATH /usr/local/lib/python3.6/site-packages/elasticdl/go/bin:$PATH
+RUN /bin/bash -c\
+ 'PYTHON_PKG_PATH=$(pip3 show elasticdl | grep "Location:" | cut -d " " -f2);\
+ echo "PATH=${PYTHON_PKG_PATH}/elasticdl/go/bin:$PATH" >> /root/.bashrc'
 
 COPY . /model_zoo
 RUN pip install -r /model_zoo/requirements.txt\