257 lines
8.5 KiB
Python
257 lines
8.5 KiB
Python
# Copyright 2015 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
import fnmatch
|
|
import os
|
|
import re
|
|
import sys
|
|
|
|
from setuptools import find_packages, setup, Command
|
|
from setuptools.command.install import install as InstallCommandBase
|
|
from setuptools.dist import Distribution
|
|
|
|
# This version string is semver compatible, but incompatible with pip.
|
|
# For pip, we will remove all '-' characters from this string, and use the
|
|
# result for pip.
|
|
_VERSION = '1.6.0-rc1'
|
|
|
|
REQUIRED_PACKAGES = [
|
|
'absl-py >= 0.1.6',
|
|
'astor >= 0.6.0',
|
|
'gast >= 0.2.0',
|
|
'grpcio >= 1.8.6',
|
|
'numpy >= 1.13.3',
|
|
'six >= 1.10.0',
|
|
'protobuf >= 3.4.0',
|
|
'tensorboard >= 1.6.0, < 1.7.0',
|
|
'termcolor >= 1.1.0',
|
|
]
|
|
|
|
project_name = 'tensorflow'
|
|
if '--project_name' in sys.argv:
|
|
project_name_idx = sys.argv.index('--project_name')
|
|
project_name = sys.argv[project_name_idx + 1]
|
|
sys.argv.remove('--project_name')
|
|
sys.argv.pop(project_name_idx)
|
|
|
|
# python3 requires wheel 0.26
|
|
if sys.version_info.major == 3:
|
|
REQUIRED_PACKAGES.append('wheel >= 0.26')
|
|
else:
|
|
REQUIRED_PACKAGES.append('wheel')
|
|
# mock comes with unittest.mock for python3, need to install for python2
|
|
REQUIRED_PACKAGES.append('mock >= 2.0.0')
|
|
|
|
# tf-nightly should depend on tb-nightly
|
|
if 'tf_nightly' in project_name:
|
|
for i, pkg in enumerate(REQUIRED_PACKAGES):
|
|
if 'tensorboard' in pkg:
|
|
REQUIRED_PACKAGES[i] = 'tb-nightly >= 1.7.0a0, < 1.8.0a0'
|
|
break
|
|
|
|
# weakref.finalize and enum were introduced in Python 3.4
|
|
if sys.version_info < (3, 4):
|
|
REQUIRED_PACKAGES.append('backports.weakref >= 1.0rc1')
|
|
REQUIRED_PACKAGES.append('enum34 >= 1.1.6')
|
|
|
|
# pylint: disable=line-too-long
|
|
CONSOLE_SCRIPTS = [
|
|
'freeze_graph = tensorflow.python.tools.freeze_graph:main',
|
|
'toco_from_protos = tensorflow.contrib.lite.toco.python.toco_from_protos:main',
|
|
'toco = tensorflow.contrib.lite.toco.python.toco_wrapper:main',
|
|
'saved_model_cli = tensorflow.python.tools.saved_model_cli:main',
|
|
# We need to keep the TensorBoard command, even though the console script
|
|
# is now declared by the tensorboard pip package. If we remove the
|
|
# TensorBoard command, pip will inappropriately remove it during install,
|
|
# even though the command is not removed, just moved to a different wheel.
|
|
'tensorboard = tensorboard.main:run_main',
|
|
]
|
|
# pylint: enable=line-too-long
|
|
|
|
# remove the tensorboard console script if building tf_nightly
|
|
if 'tf_nightly' in project_name:
|
|
CONSOLE_SCRIPTS.remove('tensorboard = tensorboard.main:run_main')
|
|
|
|
TEST_PACKAGES = [
|
|
'scipy >= 0.15.1',
|
|
]
|
|
|
|
class BinaryDistribution(Distribution):
|
|
def has_ext_modules(self):
|
|
return True
|
|
|
|
|
|
class InstallCommand(InstallCommandBase):
|
|
"""Override the dir where the headers go."""
|
|
|
|
def finalize_options(self):
|
|
ret = InstallCommandBase.finalize_options(self)
|
|
self.install_headers = os.path.join(self.install_purelib,
|
|
'tensorflow', 'include')
|
|
return ret
|
|
|
|
|
|
class InstallHeaders(Command):
|
|
"""Override how headers are copied.
|
|
|
|
The install_headers that comes with setuptools copies all files to
|
|
the same directory. But we need the files to be in a specific directory
|
|
hierarchy for -I <include_dir> to work correctly.
|
|
"""
|
|
description = 'install C/C++ header files'
|
|
|
|
user_options = [('install-dir=', 'd',
|
|
'directory to install header files to'),
|
|
('force', 'f',
|
|
'force installation (overwrite existing files)'),
|
|
]
|
|
|
|
boolean_options = ['force']
|
|
|
|
def initialize_options(self):
|
|
self.install_dir = None
|
|
self.force = 0
|
|
self.outfiles = []
|
|
|
|
def finalize_options(self):
|
|
self.set_undefined_options('install',
|
|
('install_headers', 'install_dir'),
|
|
('force', 'force'))
|
|
|
|
def mkdir_and_copy_file(self, header):
|
|
install_dir = os.path.join(self.install_dir, os.path.dirname(header))
|
|
# Get rid of some extra intervening directories so we can have fewer
|
|
# directories for -I
|
|
install_dir = re.sub('/google/protobuf_archive/src', '', install_dir)
|
|
|
|
# Copy eigen code into tensorflow/include.
|
|
# A symlink would do, but the wheel file that gets created ignores
|
|
# symlink within the directory hierarchy.
|
|
# NOTE(keveman): Figure out how to customize bdist_wheel package so
|
|
# we can do the symlink.
|
|
if 'external/eigen_archive/' in install_dir:
|
|
extra_dir = install_dir.replace('external/eigen_archive', '')
|
|
if not os.path.exists(extra_dir):
|
|
self.mkpath(extra_dir)
|
|
self.copy_file(header, extra_dir)
|
|
|
|
if not os.path.exists(install_dir):
|
|
self.mkpath(install_dir)
|
|
return self.copy_file(header, install_dir)
|
|
|
|
def run(self):
|
|
hdrs = self.distribution.headers
|
|
if not hdrs:
|
|
return
|
|
|
|
self.mkpath(self.install_dir)
|
|
for header in hdrs:
|
|
(out, _) = self.mkdir_and_copy_file(header)
|
|
self.outfiles.append(out)
|
|
|
|
def get_inputs(self):
|
|
return self.distribution.headers or []
|
|
|
|
def get_outputs(self):
|
|
return self.outfiles
|
|
|
|
|
|
def find_files(pattern, root):
|
|
"""Return all the files matching pattern below root dir."""
|
|
for path, _, files in os.walk(root):
|
|
for filename in fnmatch.filter(files, pattern):
|
|
yield os.path.join(path, filename)
|
|
|
|
|
|
matches = ['../' + x for x in find_files('*', 'external') if '.py' not in x]
|
|
|
|
so_lib_paths = [
|
|
i for i in os.listdir('.')
|
|
if os.path.isdir(i) and fnmatch.fnmatch(i, '_solib_*')
|
|
]
|
|
|
|
for path in so_lib_paths:
|
|
matches.extend(
|
|
['../' + x for x in find_files('*', path) if '.py' not in x]
|
|
)
|
|
|
|
if os.name == 'nt':
|
|
EXTENSION_NAME = 'python/_pywrap_tensorflow_internal.pyd'
|
|
else:
|
|
EXTENSION_NAME = 'python/_pywrap_tensorflow_internal.so'
|
|
|
|
headers = (list(find_files('*.h', 'tensorflow/core')) +
|
|
list(find_files('*.h', 'tensorflow/stream_executor')) +
|
|
list(find_files('*.h', 'google/protobuf_archive/src')) +
|
|
list(find_files('*', 'third_party/eigen3')) +
|
|
list(find_files('*', 'external/eigen_archive')) +
|
|
list(find_files('*.h', 'external/nsync/public')))
|
|
|
|
setup(
|
|
name=project_name,
|
|
version=_VERSION.replace('-', ''),
|
|
description='TensorFlow helps the tensors flow',
|
|
long_description='',
|
|
url='https://www.tensorflow.org/',
|
|
author='Google Inc.',
|
|
author_email='opensource@google.com',
|
|
# Contained modules and scripts.
|
|
packages=find_packages(),
|
|
entry_points={
|
|
'console_scripts': CONSOLE_SCRIPTS,
|
|
},
|
|
headers=headers,
|
|
install_requires=REQUIRED_PACKAGES,
|
|
tests_require=REQUIRED_PACKAGES + TEST_PACKAGES,
|
|
# Add in any packaged data.
|
|
include_package_data=True,
|
|
package_data={
|
|
'tensorflow': [
|
|
EXTENSION_NAME,
|
|
] + matches,
|
|
},
|
|
zip_safe=False,
|
|
distclass=BinaryDistribution,
|
|
cmdclass={
|
|
'install_headers': InstallHeaders,
|
|
'install': InstallCommand,
|
|
},
|
|
# 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',
|
|
'Programming Language :: Python :: 2.7',
|
|
'Programming Language :: Python :: 3',
|
|
'Programming Language :: Python :: 3.4',
|
|
'Programming Language :: Python :: 3.5',
|
|
'Programming Language :: Python :: 3.6',
|
|
'Topic :: Scientific/Engineering',
|
|
'Topic :: Scientific/Engineering :: Mathematics',
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence',
|
|
'Topic :: Software Development',
|
|
'Topic :: Software Development :: Libraries',
|
|
'Topic :: Software Development :: Libraries :: Python Modules',
|
|
],
|
|
license='Apache 2.0',
|
|
keywords='tensorflow tensor machine learning',)
|