diff --git a/tests/e2e/mnist_pip_requirements.txt b/tests/e2e/mnist_pip_requirements.txt
index 05ed5816..760238e5 100644
--- a/tests/e2e/mnist_pip_requirements.txt
+++ b/tests/e2e/mnist_pip_requirements.txt
@@ -1,4 +1,4 @@
-pytorch_lightning==1.9.5
+pytorch_lightning==2.4.0
 torchmetrics==0.9.1
 torchvision==0.19.0
 minio