diff --git a/pyproject.toml b/pyproject.toml
index 77813e63a..1ad37026a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ Repository = "https://github.com/open-lm-engine/lm-engine"
 
 [project.optional-dependencies]
 cuda = [
-    "torch>=2.10.0",
+    "torch==2.10.0",
 ]
 dev = [
     "pre-commit>=4.4.0",
@@ -46,12 +46,12 @@ tpu = [
 ]
 data = [
     "ray[default]>=2.52.1",
-    "torch>=2.10.0",
+    "torch==2.10.0",
     "zstandard>=0.25.0",
     "aiobotocore>=3.0.0",
 ]
 torch = [
-    "torch>=2.10.0",
+    "torch==2.10.0",
 ]
 
 [tool.hatch.build.targets.wheel]