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
+