From e8da7dc13f16a23c7a295320111bc7055bfdb3f6 Mon Sep 17 00:00:00 2001 From: misko user Date: Tue, 12 Nov 2024 21:33:16 +0000 Subject: [PATCH 1/2] add dgl to conda --- packages/env.cpu.yml | 2 ++ packages/env.gpu.yml | 4 +++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/packages/env.cpu.yml b/packages/env.cpu.yml index 985d8b7219..3182f19bcf 100644 --- a/packages/env.cpu.yml +++ b/packages/env.cpu.yml @@ -1,9 +1,11 @@ channels: - pytorch - conda-forge +- dglteam/label/th24_cpu - defaults dependencies: - cpuonly +- dgl - pytorch>=2.4 - ase - e3nn>=0.5 diff --git a/packages/env.gpu.yml b/packages/env.gpu.yml index 50b0c62314..413eb949fe 100644 --- a/packages/env.gpu.yml +++ b/packages/env.gpu.yml @@ -2,10 +2,12 @@ channels: - pytorch - nvidia - conda-forge +- dglteam/label/th24_cu121 - defaults dependencies: +- dgl - pytorch-cuda=12.1 -- pytorch>=2.4 +- pytorch==2.4.0 - ase - e3nn>=0.5 - numpy >=1.26.0,<2.0.0 From c526e5d2c8d3a5148517190e318825f33f055e73 Mon Sep 17 00:00:00 2001 From: misko user Date: Tue, 12 Nov 2024 21:58:59 +0000 Subject: [PATCH 2/2] make gpu and cpu pytorch versions fixed to the same --- packages/env.cpu.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/packages/env.cpu.yml b/packages/env.cpu.yml index 3182f19bcf..85a05cdcbc 100644 --- a/packages/env.cpu.yml +++ b/packages/env.cpu.yml @@ -6,7 +6,7 @@ channels: dependencies: - cpuonly - dgl -- pytorch>=2.4 +- pytorch==2.4.0 - ase - e3nn>=0.5 - numpy >=1.26.0,<2.0.0