From 4af77c6bf17bf868c8c649bc415fba679906bbf4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= <aedu.waelchli@gmail.com>
Date: Sat, 4 Nov 2023 15:29:37 +0100
Subject: [PATCH] Install both `lightning` and `pytorch-lightning` in docker
 image (#18898)

---
 dockers/release/Dockerfile | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/dockers/release/Dockerfile b/dockers/release/Dockerfile
index f69e0cdec15f8..ebc04f7164e5f 100644
--- a/dockers/release/Dockerfile
+++ b/dockers/release/Dockerfile
@@ -24,8 +24,6 @@ ARG LIGHTNING_VERSION=""
 
 COPY ./ /home/lightning/
 
-ENV PACKAGE_NAME=pytorch
-
 # install dependencies
 RUN \
     cd /home && \
@@ -42,7 +40,8 @@ RUN \
     # otherwise there is collision with folder name and pkg name on Pypi
     cd lightning && \
     pip install setuptools==59.5.0 && \
-    pip install '.[extra,loggers,strategies]' --no-cache-dir && \
+    PACKAGE_NAME=lightning pip install '.[extra,loggers,strategies]' --no-cache-dir && \
+    PACKAGE_NAME=pytorch pip install '.[extra,loggers,strategies]' --no-cache-dir && \
     cd .. && \
     rm -rf lightning