2020-07-07 11:05:17 +00:00
ARG PYTHON_VERSION = 3 .7
2020-09-30 12:36:02 +00:00
ARG PYTORCH_VERSION = 1 .6
2020-07-07 11:05:17 +00:00
2020-09-30 12:36:02 +00:00
FROM pytorchlightning/pytorch_lightning:base-xla-py${PYTHON_VERSION}-torch${PYTORCH_VERSION}
2020-07-02 01:44:19 +00:00
2020-07-07 11:05:17 +00:00
#SHELL ["/bin/bash", "-c"]
2020-07-02 01:44:19 +00:00
2020-09-30 12:36:02 +00:00
COPY ./ ./pytorch-lightning/
2020-07-02 01:44:19 +00:00
2020-09-30 12:36:02 +00:00
# If using this image for tests, intall more dependencies and don"t delete the source code where the tests live.
2020-07-04 15:31:12 +00:00
RUN \
2020-09-30 12:36:02 +00:00
# Install pytorch-lightning at the current PR, plus dependencies.
#pip install -r pytorch-lightning/requirements/base.txt --no-cache-dir && \
2020-07-04 15:31:12 +00:00
# drop Horovod
2020-07-07 11:05:17 +00:00
#python -c "fname = 'pytorch-lightning/requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if not line.startswith('horovod')] ; open(fname, 'w').writelines(lines)" && \
2020-09-30 12:36:02 +00:00
pip install -r pytorch-lightning/requirements/devel.txt --no-cache-dir --upgrade-strategy only-if-needed
2020-07-02 01:44:19 +00:00
2020-07-04 15:31:12 +00:00
#RUN python -c "import pytorch_lightning as pl; print(pl.__version__)"
2020-07-02 01:44:19 +00:00
2020-09-30 12:36:02 +00:00
COPY ./dockers/tpu-tests/docker-entrypoint.sh /usr/local/bin/
2020-07-02 01:44:19 +00:00
RUN chmod +x /usr/local/bin/docker-entrypoint.sh
ENTRYPOINT [ "/usr/local/bin/docker-entrypoint.sh" ]
CMD [ "bash" ]
