diff --git a/pyproject.toml b/pyproject.toml
index 0ce2e9c3..bd964e1e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,7 +38,7 @@ dependencies = [
   "torchmetrics",
   "rich==13.5.*",
   "opencv-python==4.8.0.*",
-  "torch>=2.0"
+  "torch>=2.0,<2.2"
 ]
 dynamic = ["version"]