Skip to content

GitLab

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

Install package dependencies automatically

parent ce097e1a
Show whitespace changes
Inline Side-by-side
Showing with 9 additions and 0 deletions
setup.py
View file @ 806a66bc
...@@ -9,6 +9,14 @@ VERSION = '0.1.6' ...@@ -9,6 +9,14 @@ VERSION = '0.1.6'
long_description = '''torch-vision provides DataLoaders, Pre-trained models long_description = '''torch-vision provides DataLoaders, Pre-trained models
and common transforms for torch for images and videos''' and common transforms for torch for images and videos'''
# same as ./requirements.txt
requirements = [
'numpy',
'pillow',
'six',
'torch',
]
setup_info = dict( setup_info = dict(
# Metadata # Metadata
name='torchvision', name='torchvision',
...@@ -24,6 +32,7 @@ setup_info = dict( ...@@ -24,6 +32,7 @@ setup_info = dict(
packages=find_packages(exclude=('test',)), packages=find_packages(exclude=('test',)),
zip_safe=True, zip_safe=True,
install_requires=requirements,
) )
setup(**setup_info) 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