diff --git a/pyproject.toml b/pyproject.toml
index 0a066326..b53b8131 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -14,7 +14,7 @@ dependencies = [
     "xarray",
     "ipykernel",
     "h5netcdf",
-    "torch>=2.0.0",
+    "torch>=2.0.0, <2.5.0",
     "lightning",
     "torchvision",
     "pytest",