Skip to content

Commit

Permalink
Add a setup.py script
Browse files Browse the repository at this point in the history
  • Loading branch information
lukeyeager committed Sep 21, 2016
1 parent e6ca94e commit 8264a15
Show file tree
Hide file tree
Showing 5 changed files with 67 additions and 28 deletions.
7 changes: 4 additions & 3 deletions .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
3 changes: 3 additions & 0 deletions MANIFEST.in
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
recursive-include digits/templates *
recursive-include digits/static *
recursive-include digits/standard-networks *
19 changes: 0 additions & 19 deletions requirements.txt

This file was deleted.

6 changes: 0 additions & 6 deletions requirements_test.txt

This file was deleted.

60 changes: 60 additions & 0 deletions setup.py
Original file line number Diff line number Diff line change
@@ -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'],
)

0 comments on commit 8264a15

Please sign in to comment.