diff --git a/env.yml b/env.yml
index 221d376..abb4de0 100644
--- a/env.yml
+++ b/env.yml
@@ -1,6 +1,9 @@
 name: vis149
 channels:
 - conda-forge
+- pytorch-nightly
+- pytorch
+- defaults
 dependencies:
 #- python=3.9
 #- tensorflow=2.9.1=cuda112*