diff --git a/conda_deps_linux.txt b/conda_deps_linux.txt
index 10b81169..4e1ad1e3 100644
--- 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
index 9ed70c3b..1614d743 100644
--- 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
\ No newline at end of file
+clangxx
+llvm-openmp
\ No newline at end of file
diff --git a/conda_deps_win.txt b/conda_deps_win.txt
index 20e7f9ce..4b6e5274 100644
--- a/conda_deps_win.txt
+++ b/conda_deps_win.txt
@@ -1,6 +1,7 @@
 h5py
 pip
-pytorch
+libtorch
+pytorch-cpu
 pytorch_geometric
 lightning
 torchmetrics