updated deps

torchmd · Jan 22, 2025 · a498ee7 · a498ee7
1 parent b57205b
commit a498ee7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/conda_deps_linux.txt b/conda_deps_linux.txt
@@ -1,7 +1,8 @@
 h5py
 nnpops
 pip
-pytorch
+libtorch
+pytorch-cpu
 pytorch_geometric
 lightning
 torchmetrics

diff --git a/conda_deps_osx.txt b/conda_deps_osx.txt
@@ -1,9 +1,11 @@
 h5py
 nnpops
 pip
-pytorch
+libtorch
+pytorch-cpu
 pytorch_geometric
 lightning
 torchmetrics
 tqdm
-clangxx
+clangxx
+llvm-openmp
diff --git a/conda_deps_win.txt b/conda_deps_win.txt
@@ -1,6 +1,7 @@
 h5py
 pip
-pytorch
+libtorch
+pytorch-cpu
 pytorch_geometric
 lightning
 torchmetrics