@@ -23,30 +23,30 @@ ARG LIGHTNING_VERSION=""
23
23
24
24
RUN python -c "import torch ; print(torch.__version__)" >> torch_version.info
25
25
26
- COPY ./ /workspace/lightning/
26
+ COPY ./ /workspace/pytorch- lightning/
27
27
28
28
RUN \
29
29
cd /workspace && \
30
30
# replace by specific version if asked
31
31
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 ; \
35
35
git checkout ${LIGHTNING_VERSION} ; \
36
36
git submodule update --init --recursive ; \
37
37
cd .. ; \
38
38
fi && \
39
39
# 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 . && \
45
45
46
46
# Installations \
47
47
pip install "Pillow>=8.2, !=8.3.0" "cryptography>=3.4" "py>=1.10" --no-cache-dir && \
48
48
PACKAGE_NAME=pytorch pip install './lightning[extra,loggers,strategies]' --no-cache-dir && \
49
- rm -rf lightning && \
49
+ rm -rf pytorch- lightning && \
50
50
pip list
51
51
52
52
RUN pip install jupyterlab[all] -U --no-cache-dir
0 commit comments