Skip to content

Commit 201fb4d

Browse files
awaelchlilantiga
authored andcommitted
Install both lightning and pytorch-lightning in docker image (#18898)
(cherry picked from commit 4af77c6)
1 parent 80efc01 commit 201fb4d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

dockers/release/Dockerfile

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,6 @@ ARG LIGHTNING_VERSION=""
2424

2525
COPY ./ /home/lightning/
2626

27-
ENV PACKAGE_NAME=pytorch
28-
2927
# install dependencies
3028
RUN \
3129
cd /home && \
@@ -42,7 +40,8 @@ RUN \
4240
# otherwise there is collision with folder name and pkg name on Pypi
4341
cd lightning && \
4442
pip install setuptools==59.5.0 && \
45-
pip install '.[extra,loggers,strategies]' --no-cache-dir && \
43+
PACKAGE_NAME=lightning pip install '.[extra,loggers,strategies]' --no-cache-dir && \
44+
PACKAGE_NAME=pytorch pip install '.[extra,loggers,strategies]' --no-cache-dir && \
4645
cd .. && \
4746
rm -rf lightning
4847

0 commit comments

Comments
 (0)