diff --git a/environment.yaml b/environment.yaml
index f6be819..defb5b4 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -1,10 +1,13 @@
 name: lightsheet
 
 channels:
+  - pytorch
   - conda-forge
 
 dependencies:
+  - cpuonly
   - scikit-image
   - pybdv
+  - pytorch
   - torch_em
   - z5py