diff --git a/pyproject.toml b/pyproject.toml
index 7b8e29c..f4ad917 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -54,7 +54,7 @@ mpmath = "^1.3.0"
 scipy = "^1.10.1"
 matplotlib = "^3.7.1"
 joblib = "^1.2.0"
-torch = "^2.0.0"
+torch = "==2.1.1"
 tqdm = "^4.66.1"
 
 [tool.poetry.group.dev.dependencies]