From d537485e410fcd515ade7aaac5125f040b2d1133 Mon Sep 17 00:00:00 2001 From: Jeff Yang Date: Mon, 2 Nov 2020 00:47:51 +0630 Subject: [PATCH] lock pytorch nightly version (#4469) --- environment.yml | 4 ++-- requirements.txt | 2 +- requirements/examples.txt | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/environment.yml b/environment.yml index 09eb997b78..3d59c1eeed 100644 --- 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 index 0f8423e086..d270e2bc5d 100644 --- 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 index c87d10a393..e930579b8b 100644 --- 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 \ No newline at end of file