diff --git a/setup.py b/setup.py
index 72d49f1..88b3393 100644
--- a/setup.py
+++ b/setup.py
@@ -26,7 +26,7 @@ def load_long_description():
     packages=find_packages(exclude=['tests', 'tests/*']),
     install_requires=[
         'neptune-client',
-        'pytorch_lightning == 0.9.0',
+        'pytorch_lightning == 2.4.0',
         'hydra-core >= 1.0.0',
         'transformers >= 3.0'
     ]