diff --git a/environment_cpu.yaml b/environment_cpu.yaml
index e93d7320..edc8555b 100644
--- a/environment_cpu.yaml
+++ b/environment_cpu.yaml
@@ -7,6 +7,7 @@ dependencies:
     - affogato
     - bioimageio.core >=0.5.0
     - cpuonly
+    - imagecodecs
     - python-elf
     - pytorch >=2.0
     - tensorboard
diff --git a/environment_gpu.yaml b/environment_gpu.yaml
index 6b2fdf6d..61cd105d 100644
--- a/environment_gpu.yaml
+++ b/environment_gpu.yaml
@@ -7,6 +7,7 @@ name:
 dependencies:
     - affogato
     - bioimageio.core >=0.5.0
+    - imagecodecs
     - python-elf
     - pytorch >=2.0
     - pytorch-cuda>=11.7  # you may need to update the pytorch version to match your system