diff --git a/environment.yml b/environment.yml
index b17a50f..e488177 100644
--- a/environment.yml
+++ b/environment.yml
@@ -9,9 +9,9 @@ dependencies:
   - python=3.7.13
   - pip=22.3.1
   - pytorch=1.12.1
-  - torchaudio=0.12.1
+  - torchmetrics
   - torchvision=0.13.1
   - tqdm
   - pip:
     - submodules/diff-gaussian-rasterization
-    - submodules/simple-knn
\ No newline at end of file
+    - submodules/simple-knn