diff --git a/com.unity.ml-agents/CHANGELOG.md b/com.unity.ml-agents/CHANGELOG.md
index 0efa435bc7..ae88dd1744 100755
--- a/com.unity.ml-agents/CHANGELOG.md
+++ b/com.unity.ml-agents/CHANGELOG.md
@@ -12,6 +12,7 @@ and this project adheres to
 - Fixed tensor indexing to use correct CHW layout (#6239)
 
 #### ml-agents / ml-agents-envs
+- Set the Torch version constraint to 2.8 (#6251)
 - Fixed CUDA/CPU mismatch in threaded training (#6245)
 
 ## [4.0.0] - 2025-08-15
diff --git a/ml-agents/setup.py b/ml-agents/setup.py
index 16be25d8b3..2aec4e2cd6 100644
--- a/ml-agents/setup.py
+++ b/ml-agents/setup.py
@@ -62,7 +62,7 @@ def run(self):
         "Pillow>=4.2.1",
         "protobuf>=3.6,<3.21",
         "pyyaml>=3.1.0",
-        "torch>=2.1.1",
+        "torch>=2.1.1,<=2.8.0",
         "tensorboard>=2.14",
         # adding six explicit dependency since tensorboard needs it but doesn't declare it as a dep
         "six>=1.16",