Merge pull request #29 from smartkiwi/master

Install package dependencies automatically
pytorch · Jan 19, 2017 · e869fcb · e869fcb
2 parents 048f6f5 + 2fb8447
commit e869fcb
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -9,6 +9,14 @@
 
 VERSION = '0.1.6'
 
+# same as ./requirements.txt
+requirements = [
+    'numpy',
+    'pillow',
+    'six',
+    'torch',
+]
+
 setup(
     # Metadata
     name='torchvision',
@@ -24,4 +32,5 @@
     packages=find_packages(exclude=('test',)),
 
     zip_safe=True,
+    install_requires=requirements,
 )