diff --git a/containers/env-template.yml b/containers/env-template.yml
index 064364352084810dd9461c63915f20a67cd97c18..330aeb676d6a665df3db7ade7a7f3a6b8bcfce01 100644
--- a/containers/env-template.yml
+++ b/containers/env-template.yml
@@ -3,6 +3,7 @@
 channels:
   - conda-forge
   - defaults
+  - pytorch
 dependencies:
   - pip
   - setuptools
@@ -31,6 +32,7 @@ dependencies:
   - cython
   - pytables
   - pytorch
+  - cpuonly
   - python-lalsimulation
   - bilby.cython
   - pyseobnr