From 5966654052b24d99e4bfbcf1b59faae8a75db1fd Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 24 May 2020 13:28:03 -0700 Subject: [PATCH] Update requirements so PyTorch 1.4 is min, add separate sotabench reqs to force 1.4 and prevent error with 1.5 --- requirements-sotabench.txt | 3 +++ requirements.txt | 4 ++-- sotabench_setup.sh | 5 +++++ 3 files changed, 10 insertions(+), 2 deletions(-) create mode 100644 requirements-sotabench.txt create mode 100755 sotabench_setup.sh diff --git a/requirements-sotabench.txt b/requirements-sotabench.txt new file mode 100644 index 00000000..2e4392dd --- /dev/null +++ b/requirements-sotabench.txt @@ -0,0 +1,3 @@ +torch==1.4.0 +torchvision==0.5.0 +pyyaml diff --git a/requirements.txt b/requirements.txt index f05f9812..2d29a27c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,3 @@ -torch>=1.2.0 -torchvision>=0.4.0 +torch>=1.4.0 +torchvision>=0.5.0 pyyaml diff --git a/sotabench_setup.sh b/sotabench_setup.sh new file mode 100755 index 00000000..693ea752 --- /dev/null +++ b/sotabench_setup.sh @@ -0,0 +1,5 @@ +#!/usr/bin/env bash +source /workspace/venv/bin/activate + +pip install -r requirements-sotabench.txt +