diff --git a/pyproject.toml b/pyproject.toml
index dfdb3859..6292ae02 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,7 +16,7 @@ dependencies = [
     "jsonschema",
     "omegaconf",
     "python-hostlist",
-    "torch",
+    "torch==2.3.1",
     "vesin",
 ]