cube-studio/job-template/job/tf_distributed_train/Dockerfile
2022-02-26 22:36:02 +08:00

14 lines
710 B
Docker

FROM python:3.7.8
RUN wget https://github.com/wercker/stern/releases/download/1.11.0/stern_linux_amd64 && chmod +x stern_linux_amd64 && mv stern_linux_amd64 /usr/bin/stern
RUN apt update && apt install -y --force-yes --no-install-recommends vim apt-transport-https gnupg2 ca-certificates-java rsync jq wget git dnsutils iputils-ping net-tools curl locales zip software-properties-common && apt-get autoclean
RUN python -m pip install --upgrade pip && \
pip install requests && \
pip install kubernetes psutil pysnooper
COPY job/pkgs /app/job/pkgs
COPY job/tf_distributed_train/*.py /app/job/tf_distributed_train/
WORKDIR /app
ENTRYPOINT ["python", "-m", "job.tf_distributed_train.tfjob_launcher"]