1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
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',
)
|
