diff --git a/packages/env.cpu.yml b/packages/env.cpu.yml index 985d8b721..a9cba3700 100644 --- a/packages/env.cpu.yml +++ b/packages/env.cpu.yml @@ -4,7 +4,7 @@ channels: - defaults dependencies: - cpuonly -- pytorch>=2.4 +- pytorch==2.4.0 - ase - e3nn>=0.5 - numpy >=1.26.0,<2.0.0 diff --git a/packages/env.gpu.yml b/packages/env.gpu.yml index 50b0c6231..0e199c055 100644 --- a/packages/env.gpu.yml +++ b/packages/env.gpu.yml @@ -5,7 +5,7 @@ channels: - defaults dependencies: - pytorch-cuda=12.1 -- pytorch>=2.4 +- pytorch==2.4.0 - ase - e3nn>=0.5 - numpy >=1.26.0,<2.0.0 diff --git a/packages/fairchem-core/pyproject.toml b/packages/fairchem-core/pyproject.toml index dfd5c671a..5081f624b 100644 --- a/packages/fairchem-core/pyproject.toml +++ b/packages/fairchem-core/pyproject.toml @@ -9,7 +9,7 @@ license = {text = "MIT License"} dynamic = ["version", "readme"] requires-python = ">=3.9, <3.13" dependencies = [ - "torch>=2.4", + "torch==2.4", "numpy >=1.26.0, <2.0.0", "lmdb", "ase",