diff --git a/requirements.txt b/requirements.txt
index 1d4be14..bbd5c5d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,6 +7,6 @@ tqdm
 pandas
 numpy
 torchsummaryX>=2.0
-pytorch_lightning==0.7.1
+pytorch_lightning==2.4.0
 tensorboardX
 optuna