swh:1:snp:a03238469985e0810f5e3a598c698eb19561f28e
Tip revision: d14960fb244e310e3468ea20b6df2e09e20b480d authored by Luke Yeager on 14 October 2016, 22:34:39 UTC
Mark v5.0.0-rc.1
Mark v5.0.0-rc.1
Tip revision: d14960f
setup.py
#!/usr/bin/env python2
# Copyright (c) 2016, NVIDIA CORPORATION. All rights reserved.
import os.path
import setuptools
LOCAL_DIR = os.path.dirname(os.path.abspath(__file__))
# Get current __version__
execfile(os.path.join(LOCAL_DIR, 'digits', 'version.py'))
# Get requirements
requirements = []
with open(os.path.join(LOCAL_DIR, 'requirements.txt'), 'r') as infile:
for line in infile:
line = line.strip()
if line and not line[0] == '#': # ignore comments
requirements.append(line)
# Get test requirements
test_requirements = []
with open(os.path.join(LOCAL_DIR, 'requirements_test.txt'), 'r') as infile:
for line in infile:
line = line.strip()
if line and not line[0] == '#': # ignore comments
test_requirements.append(line)
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=requirements,
extras_require={ 'test': test_requirements },
scripts=['digits-devserver'],
)