40 lines
1.4 KiB
Docker
40 lines
1.4 KiB
Docker
ARG PYTHON_VERSION=3.7
|
|
|
|
FROM pytorchlightning/pytorch_lightning:XLA-extras-py${PYTHON_VERSION}
|
|
|
|
# Build args.
|
|
ARG GITHUB_REF=refs/heads/master
|
|
ARG TEST_IMAGE=0
|
|
|
|
# This Dockerfile installs pytorch/xla 3.7 wheels. There are also 3.6 wheels available; see below.
|
|
|
|
#SHELL ["/bin/bash", "-c"]
|
|
|
|
# Install pytorch-lightning at the current PR, plus dependencies.
|
|
RUN git clone https://github.com/PyTorchLightning/pytorch-lightning.git && \
|
|
cd pytorch-lightning && \
|
|
echo $GITHUB_REF && \
|
|
git fetch origin $GITHUB_REF:CI && \
|
|
git checkout CI && \
|
|
pip install --requirement ./requirements/base.txt --no-cache-dir
|
|
|
|
# If using this image for tests, intall more dependencies and don"t delete
|
|
# the source code where the tests live.
|
|
RUN \
|
|
# drop Horovod
|
|
#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)" && \
|
|
#pip install --requirement pytorch-lightning/requirements/extra.txt --no-cache-dir && \
|
|
if [ $TEST_IMAGE -eq 1 ] ; then \
|
|
pip install --requirement pytorch-lightning/requirements/test.txt --no-cache-dir ; \
|
|
else \
|
|
rm -rf pytorch-lightning ; \
|
|
fi
|
|
|
|
#RUN python -c "import pytorch_lightning as pl; print(pl.__version__)"
|
|
|
|
COPY docker-entrypoint.sh /usr/local/bin/
|
|
RUN chmod +x /usr/local/bin/docker-entrypoint.sh
|
|
|
|
ENTRYPOINT ["/usr/local/bin/docker-entrypoint.sh"]
|
|
CMD ["bash"]
|