From 806a66bc9dcfee59e4a304d5069ae5b88a1179a6 Mon Sep 17 00:00:00 2001
From: smartkiwi <smartkiwi@gmail.com>
Date: Wed, 18 Jan 2017 16:30:57 -0700
Subject: [PATCH] Install package dependencies automatically

---
 setup.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/setup.py b/setup.py
index 269a1638b33..2ae08b0d3ff 100644
--- a/setup.py
+++ b/setup.py
@@ -9,6 +9,14 @@
 long_description = '''torch-vision provides DataLoaders, Pre-trained models
 and common transforms for torch for images and videos'''
 
+# same as ./requirements.txt
+requirements = [
+    'numpy',
+    'pillow',
+    'six',
+    'torch',
+]
+
 setup_info = dict(
     # Metadata
     name='torchvision',
@@ -24,6 +32,7 @@
     packages=find_packages(exclude=('test',)),
 
     zip_safe=True,
+    install_requires=requirements,
 )
 
 setup(**setup_info)