Update environment.yml

Updated pytorch version.
Waller-Lab · Jun 15, 2022 · b9d3c4f · b9d3c4f
1 parent 6d44b34
commit b9d3c4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/environment.yml b/environment.yml
@@ -5,6 +5,6 @@ dependencies:
   - python=3.6
   - anaconda
   - torchvision
-  - pytorch
+  - pytorch=1.6.0
   - cudatoolkit=10.2