Commit 806a66bc authored by smartkiwi's avatar smartkiwi Committed by GitHub
Browse files

Install package dependencies automatically

parent ce097e1a
......@@ -9,6 +9,14 @@ VERSION = '0.1.6'
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 @@ setup_info = dict(
packages=find_packages(exclude=('test',)),
zip_safe=True,
install_requires=requirements,
)
setup(**setup_info)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment