diff --git a/env.yml b/env.yml
index 277a4ce..572d5b0 100644
--- 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