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',
+    )