summaryrefslogtreecommitdiff
path: root/requirements.txt
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2020-12-18 22:38:25 +0800
committerJordan Gong <jordan.gong@protonmail.com>2020-12-18 22:38:25 +0800
commitdaad9a2ada5fb096886a4c1a4a611ce46a5979d5 (patch)
treed8c2095e8bcc74892df7c0abd71a6c7ee1dd87dc /requirements.txt
parenta08c7f31d6a9f499be9d13a1059ffc06aeb75bbf (diff)
Add torch and numpy to requirements.txt
Diffstat (limited to 'requirements.txt')
-rw-r--r--requirements.txt3
1 files changed, 3 insertions, 0 deletions
diff --git a/requirements.txt b/requirements.txt
index e69de29..46fb677 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -0,0 +1,3 @@
+torch~=1.7.1
+torchvision~=0.8.0a0+ecf4e9c
+numpy~=1.19.4 \ No newline at end of file