From dc997282bfc2e56c7168692285523cbc6fe92375 Mon Sep 17 00:00:00 2001 From: Constantin Pape Date: Wed, 27 Sep 2023 17:28:21 +0200 Subject: [PATCH 1/2] Update environment_cpu.yaml --- environment_cpu.yaml | 1 + 1 file changed, 1 insertion(+) 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 From 6cfaad70dbc4c6f3b2caf772dbcb7da75d7222ec Mon Sep 17 00:00:00 2001 From: Constantin Pape Date: Wed, 27 Sep 2023 17:29:14 +0200 Subject: [PATCH 2/2] Update environment_gpu.yaml --- environment_gpu.yaml | 1 + 1 file changed, 1 insertion(+) 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