pin pytorch at 2_4_0 (#928)

* pin pytorch at 2_4_0 * also fix pypi package --------- Co-authored-by: misko user <misko@submit-1.fair-aws-h100-2.hpcaas>
FAIR-Chem · Dec 4, 2024 · 816cf00 · 816cf00
1 parent 3f695ef
commit 816cf00
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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",