From b9d3c4fe574a81d4fb0c2acb79d643eedffd868a Mon Sep 17 00:00:00 2001
From: monakhova <monakhova@berkeley.edu>
Date: Wed, 15 Jun 2022 11:23:38 -0700
Subject: [PATCH] Update environment.yml

Updated pytorch version.
---
 environment.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/environment.yml b/environment.yml
index 6720cf4..e31afff 100644
--- a/environment.yml
+++ b/environment.yml
@@ -5,6 +5,6 @@ dependencies:
   - python=3.6
   - anaconda
   - torchvision
-  - pytorch
+  - pytorch=1.6.0
   - cudatoolkit=10.2