diff --git a/environment.yml b/environment.yml
index f11c80f..af8c0b6 100644
--- a/environment.yml
+++ b/environment.yml
@@ -4,8 +4,8 @@ channels:
     - pytorch
     - defaults
 dependencies:
-  - python=3.6
-  - pytorch
+  - python>=3.6
+  - pytorch>=1.6
   - torchvision
   - pip
   - pip: