diff options
author | Manjunath Kudlur <keveman@gmail.com> | 2015-11-06 16:27:58 -0800 |
---|---|---|
committer | Manjunath Kudlur <keveman@gmail.com> | 2015-11-06 16:27:58 -0800 |
commit | f41959ccb2d9d4c722fe8fc3351401d53bcf4900 (patch) | |
tree | ef0ca22cb2a5ac4bdec9d080d8e0788a53ed496d /tensorflow/tools/pip_package/setup.py |
TensorFlow: Initial commit of TensorFlow library.
TensorFlow is an open source software library for numerical computation
using data flow graphs.
Base CL: 107276108
Diffstat (limited to 'tensorflow/tools/pip_package/setup.py')
-rw-r--r-- | tensorflow/tools/pip_package/setup.py | 79 |
1 files changed, 79 insertions, 0 deletions
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py new file mode 100644 index 0000000000..e7f9ecdc71 --- /dev/null +++ b/tensorflow/tools/pip_package/setup.py @@ -0,0 +1,79 @@ +import fnmatch +import os +from setuptools import find_packages, setup, Extension +from setuptools.dist import Distribution + +_VERSION = '0.5.0' + +REQUIRED_PACKAGES = [ + 'numpy', + 'six >= 1.10.0', + 'virtualenvwrapper', +] + +# pylint: disable=line-too-long +CONSOLE_SCRIPTS = [ + 'tensorboard = tensorflow.tensorboard.tensorboard:main', + 'tensorflow_model_cifar10_train = tensorflow.models.image.cifar10.cifar10_train:main', + 'tensorflow_model_cifar10_multi_gpu_train = tensorflow.models.image.cifar10.cifar10_multi_gpu_train:main', + 'tensorflow_model_cifar10_eval = tensorflow.models.image.cifar10.cifar10_eval:main', + 'tensorflow_model_mnist_convolutional = tensorflow.models.image.mnist.convolutional:main', +] +# pylint: enable=line-too-long + +TEST_PACKAGES = [ + 'scipy >= 0.15.1', +] + +class BinaryDistribution(Distribution): + def is_pure(self): + return False + +matches = [] +for root, dirnames, filenames in os.walk('external'): + for filename in fnmatch.filter(filenames, '*'): + matches.append(os.path.join(root, filename)) + +matches = ['../' + x for x in matches if '.py' not in x] + +setup( + name='tensorflow', + version=_VERSION, + description='TensorFlow helps the tensors flow', + long_description='', + url='http://tensorflow.com/', + author='Google Inc.', + author_email='opensource@google.com', + # Contained modules and scripts. + packages=find_packages(), + entry_points={ + 'console_scripts': CONSOLE_SCRIPTS, + }, + install_requires=REQUIRED_PACKAGES, + tests_require=REQUIRED_PACKAGES + TEST_PACKAGES, + # Add in any packaged data. + include_package_data=True, + package_data={ + 'tensorflow': ['python/_pywrap_tensorflow.so', + 'tensorboard/dist/index.html', + 'tensorboard/dist/tf-tensorboard.html', + 'tensorboard/lib/css/global.css', + ] + matches, + }, + zip_safe=False, + distclass=BinaryDistribution, + # PyPI package information. + classifiers=[ + 'Development Status :: 4 - Beta', + 'Intended Audience :: Developers', + 'Intended Audience :: Education', + 'Intended Audience :: Science/Research', + 'License :: OSI Approved :: Apache Software License', + 'Programming Language :: Python :: 2.7', + 'Topic :: Scientific/Engineering :: Mathematics', + 'Topic :: Software Development :: Libraries :: Python Modules', + 'Topic :: Software Development :: Libraries', + ], + license='Apache 2.0', + keywords='tensorflow tensor machine learning', + ) |