diff --git a/environment.yml b/environment.yml
index c968d26b..b15e0824 100644
--- a/environment.yml
+++ b/environment.yml
@@ -9,7 +9,7 @@ dependencies:
   # - bioimage.core via submodule
   - python 3.9.*
   - numpy >=1.21,<2
-  - grpcio=1.44 # protobuf 5 requires protoc version > 3.19.0 that requires grpcio >= 1.44
+  - grpcio=1.49.1 # protobuf 5 requires protoc version > 3.19.0 that requires grpcio >= 1.44
   - marshmallow-union
   - marshmallow=3.12.*
   - marshmallow-jsonschema
@@ -20,6 +20,7 @@ dependencies:
   - scipy
   - typing-extensions
   - xarray
+  - pytorch-3dunet
   - setuptools
   - pip
 
diff --git a/setup.py b/setup.py
index 7aebd69c..8d9c03b9 100644
--- a/setup.py
+++ b/setup.py
@@ -33,6 +33,7 @@
         "numpy<2",  # pytorch 2.2.2-py3.9_0 for macos is compiled with numpy 1.*
         "protobuf",
         "pydantic>=2.7.0,<2.10",
+        "pytorch-3dunet",
         "pyyaml",
         "xarray",
     ],