Merge pull request #277 from lappazos/new_lightning_version

New pytorch lightning version

Author: dimapihtar

Dockerfile

@@ -185,7 +185,7 @@ RUN pip install --no-cache-dir wandb==0.15.3 \
         'ipython>=7.31.1' \
         'torchmetrics==0.9.1'
 
-RUN pip install pytorch_lightning==2.0.7
+RUN pip install pytorch_lightning==2.2.1
 # Copy FasterTransformer
 #COPY --from=ft_builder /workspace/FasterTransformer FasterTransformer
 
@@ -204,4 +204,4 @@ ARG NVIDIA_BUILD_ID
 ENV NVIDIA_BUILD_ID ${NVIDIA_BUILD_ID:-<unknown>}
 LABEL com.nvidia.build.id="${NVIDIA_BUILD_ID}"
 ARG NVIDIA_BUILD_REF
-LABEL com.nvidia.build.ref="${NVIDIA_BUILD_REF}"
+LABEL com.nvidia.build.ref="${NVIDIA_BUILD_REF}"