Added missing requirements.

sharif1093 · Mar 5, 2019 · f783a13 · f783a13
1 parent d1bb788
commit f783a13
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -2,3 +2,6 @@ colorama>=0.4.1
 psutil>=5.5.1
 pyyaml>=3.13
 sphinxcontrib-programoutput>=0.13
+numpy>=1.15.4
+torch>=1.0.0
+visdom>=0.1.8.8
diff --git a/setup.py b/setup.py
@@ -10,7 +10,9 @@
     # long_description_content_type="text/markdown",
     url="https://github.com/sharif1093/digideep",
     install_requires=[
-        "colorama>=0.4.1", "psutil>=5.5.1", "pyyaml>=3.13", "sphinxcontrib-programoutput>=0.13"
+        "colorama>=0.4.1", "psutil>=5.5.1", "pyyaml>=3.13",
+        "sphinxcontrib-programoutput>=0.13", "numpy>=1.15.4",
+        "torch>=1.0.0", "visdom>=0.1.8.8"
     ],
     packages=setuptools.find_packages(),
     license="",