diff --git a/pyproject.toml b/pyproject.toml
index 1fb4a9e..6b12e0f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,7 +7,7 @@ license = "MIT"
 
 [tool.poetry.dependencies]
 python = "^3.7"
-pytorch-lightning = "^1.6.0"
+pytorch-lightning = "^2.0.0"
 torch = "^1.8.0"
 
 [tool.poetry.group.dev.dependencies]