From e32e3575d8ef30414b7b607e607c44a29980f2b1 Mon Sep 17 00:00:00 2001
From: Prathik Rao <prathik.rao@gmail.com>
Date: Tue, 13 Aug 2024 20:04:56 -0700
Subject: [PATCH] pin pytorch lightning version for training CI (#21731)

### Description
<!-- Describe your changes. -->

Pins pytorch-lightning package to version 2.3.3 since version >=2.4.0
requires torch > 2.1.0 which is not compatible with cu118.


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->

ORT 1.19 Release Preparation
---
 .../docker/scripts/training/ortmodule/stage2/requirements.txt   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt
index d7fab6a1c8a27..3b13a51f18e27 100644
--- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt
+++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt
@@ -8,7 +8,7 @@ rsa==4.9
 tensorboard==2.13.0
 h5py
 wget
-pytorch-lightning
+pytorch-lightning==2.3.3
 deepspeed==0.9.0
 fairscale==0.4.6
 parameterized>=0.8.1