Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

CI: update TPU docker (#14448)

This commit is contained in:
Jirka Borovec 2022-08-30 21:17:38 +02:00 committed by GitHub
parent 6188593abd
commit 74304db6f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dockers/tpu-tests/Dockerfile
View File

@ -19,16 +19,16 @@ FROM pytorchlightning/pytorch_lightning:base-xla-py${PYTHON_VERSION}-torch${PYTO
LABEL maintainer="Lightning-AI <https://github.com/Lightning-AI>" LABEL maintainer="Lightning-AI <https://github.com/Lightning-AI>"
COPY ./ ./pytorch-lightning/ COPY ./ ./lightning/
# Pull the legacy checkpoints # Pull the legacy checkpoints
RUN cd pytorch-lightning && \ RUN cd lightning && \
bash .actions/pull_legacy_checkpoints.sh bash .actions/pull_legacy_checkpoints.sh
RUN \ RUN \
pip install -q fire && \ pip install -q fire && \
# drop unnecessary packages # drop unnecessary packages
pip install -r pytorch-lightning/requirements/pytorch/devel.txt --no-cache-dir pip install -r lightning/requirements/pytorch/devel.txt --no-cache-dir
COPY ./dockers/tpu-tests/docker-entrypoint.sh /usr/local/bin/ COPY ./dockers/tpu-tests/docker-entrypoint.sh /usr/local/bin/
RUN chmod +x /usr/local/bin/docker-entrypoint.sh RUN chmod +x /usr/local/bin/docker-entrypoint.sh

2
dockers/tpu-tests/tpu_test_cases.jsonnet
View File

@ -22,7 +22,7 @@ local tputests = base.BaseTest {
||| |||
source ~/.bashrc source ~/.bashrc
conda activate lightning conda activate lightning
mkdir -p /home/runner/work/pytorch-lightning && cd /home/runner/work/pytorch-lightning mkdir -p /home/runner/work/lightning && cd /home/runner/work/lightning
git clone https://github.com/Lightning-AI/lightning.git git clone https://github.com/Lightning-AI/lightning.git
cd lightning cd lightning
echo $PWD echo $PWD