diff --git a/.travis.yml b/.travis.yml index 27a6a0fc7..041c09706 100644 --- a/.travis.yml +++ b/.travis.yml @@ -44,9 +44,10 @@ install: # build torch - travis_wait ./scripts/travis/install-torch-wrapper.sh ~/torch $(pwd)/torch-install-log.txt - # digits requirements - - pip install -r requirements.txt - - pip install -r requirements_test.txt + # python packages + - pip install . + - pip install digits[test] + - pip install coveralls before_script: - export CAFFE_ROOT=~/caffe diff --git a/MANIFEST.in b/MANIFEST.in new file mode 100644 index 000000000..18dc420c3 --- /dev/null +++ b/MANIFEST.in @@ -0,0 +1,3 @@ +recursive-include digits/templates * +recursive-include digits/static * +recursive-include digits/standard-networks * diff --git a/requirements.txt b/requirements.txt deleted file mode 100644 index d4e39ea77..000000000 --- a/requirements.txt +++ /dev/null @@ -1,19 +0,0 @@ -Pillow>=2.3.0,<=3.1.2 -numpy>=1.8.1,<=1.11.0 -scipy>=0.13.3,<=0.17.0 -protobuf>=2.5.0,<=2.6.1 -six>=1.5.2,<=1.10.0 -requests>=2.2.1,<=2.9.1 -gevent>=1.0,<=1.1.0 -gevent-websocket==0.9.3 -Flask==0.10.1 -Flask-WTF>=0.11,<=0.12 -wtforms>=2.0,<=2.1 -Flask-SocketIO==2.6 -gunicorn>=17.5,<=19.4.5 -setuptools>=3.3,<=20.7.0 -lmdb==0.87 -h5py>=2.2.1,<=2.6.0 -pydot>=1.0.28,<=1.0.29 -psutil>=1.2.1,<=3.4.2 -matplotlib>=1.3.1,<=1.5.1 diff --git a/requirements_test.txt b/requirements_test.txt deleted file mode 100644 index eab562217..000000000 --- a/requirements_test.txt +++ /dev/null @@ -1,6 +0,0 @@ -nose>=1.3.1,<=1.3.7 -mock>=1.0.1,<=1.3.0 -beautifulsoup4==4.4.1 -selenium>=2.25.0,<=2.48.0 -coverage==3.7.1 -coveralls==1.1 diff --git a/setup.py b/setup.py new file mode 100644 index 000000000..09a63365d --- /dev/null +++ b/setup.py @@ -0,0 +1,60 @@ +#!/usr/bin/env python2 +# Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved. + +import os.path +import setuptools + +# Get current __version__ +execfile(os.path.join(os.path.abspath(os.path.dirname(__file__)), 'digits', 'version.py')) + +setuptools.setup( + name='digits', + version=__version__, + description="NVIDIA's Deep Learning GPU Training System", + url='https://developer.nvidia.com/digits', + author='DIGITS Development Team', + author_email='digits@nvidia.com', + license='BSD', + classifiers=[ + 'Framework :: Flask', + 'License :: OSI Approved :: BSD License', + 'Programming Language :: Python :: 2.7', + 'Programming Language :: Python :: 2 :: Only', + 'Topic :: Scientific/Engineering :: Artificial Intelligence', + ], + keywords='nvidia digits', + packages=setuptools.find_packages(), + include_package_data=True, + zip_safe=False, + install_requires=[ + 'Flask==0.10.1', + 'Flask-SocketIO==2.6', + 'Flask-WTF>=0.11,<=0.12', + 'gevent>=1.0,<=1.1.0', + 'gevent-websocket==0.9.3', + 'gunicorn>=17.5,<=19.4.5', + 'h5py>=2.2.1,<=2.6.0', + 'lmdb==0.87', + 'matplotlib>=1.3.1,<=1.5.1', + 'numpy>=1.8.1,<=1.11.0', + 'Pillow>=2.3.0,<=3.1.2', + 'protobuf>=2.5.0,<=2.6.1', + 'psutil>=1.2.1,<=3.4.2', + 'pydot>=1.0.28,<=1.0.29', + 'requests>=2.2.1,<=2.9.1', + 'scipy>=0.13.3,<=0.17.0', + 'setuptools>=3.3,<=20.7.0', + 'six>=1.5.2,<=1.10.0', + 'wtforms>=2.0,<=2.1', + ], + extras_require={ + 'test': [ + 'beautifulsoup4==4.4.1', + 'coverage==3.7.1', + 'mock>=1.0.1,<=1.3.0', + 'nose>=1.3.1,<=1.3.7', + ] + }, + scripts=['digits-server', 'digits-devserver'], +) +