diff --git a/environment.yaml b/environment.yaml
index f36b0e17d..dd6f7b2af 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -13,6 +13,7 @@ dependencies:
     - albumentations==0.4.3
     - opencv-python==4.1.2.30
     - pudb==2019.2
+    - torchmetrics==v0.6.0
     - imageio==2.9.0
     - imageio-ffmpeg==0.4.2
     - pytorch-lightning==1.4.2
@@ -24,4 +25,4 @@ dependencies:
     - transformers==4.3.1
     - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
     - -e git+https://github.com/openai/CLIP.git@main#egg=clip
-    - -e .
\ No newline at end of file
+    - -e .