Update env.yml

ucsd-ets · Jan 30, 2024 · 9f7b7b9 · 9f7b7b9
1 parent 4504e36
commit 9f7b7b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/env.yml b/env.yml
@@ -26,8 +26,8 @@ dependencies:
   - networkx
   - scipy
 #  - ninja
-  - torch==1.8.0+cu111
-  - torchvision==0.9.0+cu111
+  - torch==1.8.0 #+cu111
+  - torchvision==0.9.0 #+cu111
   - torchaudio==0.8.0
 #  - tensorflow==2.12.*
 #  - nvidia-cudnn-cu11==8.6.0.163