diff --git a/setup.py b/setup.py index bd8bf649e41..c26013fe4cb 100644 --- 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, )