@@ -23,30 +23,30 @@ ARG LIGHTNING_VERSION=""
2323
2424RUN python -c "import torch ; print(torch.__version__)" >> torch_version.info
2525
26- COPY ./ /workspace/lightning/
26+ COPY ./ /workspace/pytorch- lightning/
2727
2828RUN \
2929 cd /workspace && \
3030 # replace by specific version if asked
3131 if [ ! -z "$LIGHTNING_VERSION" ] ; then \
32- rm -rf lightning ; \
33- git clone https://github.com/Lightning-AI/lightning.git ; \
34- cd lightning ; \
32+ rm -rf pytorch- lightning ; \
33+ git clone https://github.com/Lightning-AI/pytorch- lightning.git ; \
34+ cd pytorch- lightning ; \
3535 git checkout ${LIGHTNING_VERSION} ; \
3636 git submodule update --init --recursive ; \
3737 cd .. ; \
3838 fi && \
3939# save the examples \
40- ls -lh lightning/ && \
41- rm -rf lightning/.git && \
42- rm -rf lightning/_notebooks/.git && \
43- mv lightning/_notebooks/.notebooks/ notebooks && \
44- cp -r lightning/*examples . && \
40+ ls -lh pytorch- lightning/ && \
41+ rm -rf pytorch- lightning/.git && \
42+ rm -rf pytorch- lightning/_notebooks/.git && \
43+ mv pytorch- lightning/_notebooks/.notebooks/ notebooks && \
44+ cp -r pytorch- lightning/*examples . && \
4545
4646# Installations \
4747 pip install "Pillow>=8.2, !=8.3.0" "cryptography>=3.4" "py>=1.10" --no-cache-dir && \
4848 PACKAGE_NAME=pytorch pip install './lightning[extra,loggers,strategies]' --no-cache-dir && \
49- rm -rf lightning && \
49+ rm -rf pytorch- lightning && \
5050 pip list
5151
5252RUN pip install jupyterlab[all] -U --no-cache-dir
0 commit comments