Add torch to requirements

fepegar · Dec 19, 2019 · 4c2e91f · 4c2e91f
1 parent c16fa87
commit 4c2e91f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -16,6 +16,7 @@
     'nibabel',
     'numpy',
     'SimpleITK',
+    'torch>=1.1',
     'tqdm',
 ]