Merge pull request #109 from jaeyeol816/fix_build_err

specified pytorch version to fix build error
sxyu · Feb 2, 2023 · ee80e2c · ee80e2c
2 parents b63b5dc + a1cae70
commit ee80e2c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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