diff --git a/pyproject.toml b/pyproject.toml index 9c6f213cfd..e95f21e2ea 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -167,7 +167,7 @@ pin_pytorch_cpu = [ "torch; platform_machine=='x86_64' and platform_system == 'Darwin'", ] pin_pytorch_gpu = [ - "torch>=2.7,<2.10", + "torch==2.10.0", ] pin_jax = [ "jax==0.5.0;python_version>='3.10'",