lock pytorch nightly version (#4469)

Lightning-AI · Nov 1, 2020 · d537485 · d537485
1 parent 839813e
commit d537485
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/environment.yml b/environment.yml
@@ -26,7 +26,7 @@ dependencies:
     - python>=3.6
     - pip>20.1
     - numpy>=1.16.4
-    - pytorch>=1.3
+    - pytorch>=1.3,<1.8
     - future>=0.17.1
     - PyYAML>=5.1
     - tqdm>=4.41.0
@@ -41,7 +41,7 @@ dependencies:
     - torchtext>=0.3.1
 
     # Examples
-    - torchvision>=0.4.1
+    - torchvision>=0.4.1,<0.9.0
 
     - pip:
         - test-tube>=0.7.5

diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 # the default package dependencies
 
 numpy>=1.16.4
-torch>=1.3
+torch>=1.3,<1.8
 future>=0.17.1  # required for builtins in setup.py
 # pyyaml>=3.13
 PyYAML>=5.1  # OmegaConf requirement >=5.1

diff --git a/requirements/examples.txt b/requirements/examples.txt
@@ -1,2 +1,2 @@
-torchvision>=0.4.1
+torchvision>=0.4.1,<0.9.0
 gym>=0.17.0