diff --git a/env_cpu.yml b/env_cpu.yml
index f1fbba3..9b4ebe2 100644
--- a/env_cpu.yml
+++ b/env_cpu.yml
@@ -27,5 +27,6 @@ dependencies:
     - opencv-python
     - timm==1.0.9
     - pynrrd
+    - iopath>=0.1.10
     - git+https://github.com/facebookresearch/segment-anything.git
     - git+https://github.com/juglab/featureforest.git
diff --git a/env_gpu.yml b/env_gpu.yml
index 6b30f60..6663f57 100644
--- a/env_gpu.yml
+++ b/env_gpu.yml
@@ -29,5 +29,6 @@ dependencies:
     - opencv-python
     - timm==1.0.9
     - pynrrd
+    - iopath>=0.1.10
     - git+https://github.com/facebookresearch/segment-anything.git
     - git+https://github.com/juglab/featureforest.git
diff --git a/pyproject.toml b/pyproject.toml
index f55c63b..5a61892 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,6 +57,7 @@ dependencies = [
     "napari",
     "h5py",
     "pooch",
+    "iopath>=0.1.10",
     "timm==1.0.9",
     "segment-anything-py",
     "sam2 @ git+https://github.com/facebookresearch/sam2.git"