diff --git a/environment.yml b/environment.yml
index eb2ac71a..a8f6790e 100644
--- a/environment.yml
+++ b/environment.yml
@@ -22,7 +22,7 @@ dependencies:
       - moviepy
       - matplotlib
       - scipy>=1.6.0
-  - pytorch
+  - pytorch=1.11.0
   - torchvision
   - cudatoolkit
   - tqdm