experiments
/
STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Clarify why we have *-gpu package. 

Fix https://github.com/tensorflow/tensorflow/issues/39581

PiperOrigin-RevId: 311782903
Change-Id: If002f2d2b112012e1c75e0c16f7a922546a9bba5
This commit is contained in:
Yifei Feng 2020-05-15 12:41:50 -07:00 committed by TensorFlower Gardener
parent 2db0d85d05
commit 08968c30dc
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/tools/pip_package/setup.py
View File

@ -43,8 +43,6 @@ from setuptools import setup
from setuptools.command.install import install as InstallCommandBase from setuptools.command.install import install as InstallCommandBase
from setuptools.dist import Distribution from setuptools.dist import Distribution
DOCLINES = __doc__.split('\n')
# This version string is semver compatible, but incompatible with pip. # This version string is semver compatible, but incompatible with pip.
# For pip, we will remove all '-' characters from this string, and use the # For pip, we will remove all '-' characters from this string, and use the
# result for pip. # result for pip.
@ -93,6 +91,16 @@ if 'tf_nightly' in project_name:
elif 'tensorflow_estimator' in pkg: elif 'tensorflow_estimator' in pkg:
REQUIRED_PACKAGES[i] = 'tf-estimator-nightly' REQUIRED_PACKAGES[i] = 'tf-estimator-nightly'
DOCLINES = __doc__.split('\n')
if project_name.endswith('-gpu'):
project_name_no_gpu = project_name[:-len('-gpu')]
_GPU_PACKAGE_NOTE = 'Note that %s package by default supports both CPU and '\
'GPU. %s has the same content and exists solely for backward '\
'compatiblity. Please migrate to %s for GPU support.'\
% (project_name_no_gpu, project_name, project_name_no_gpu)
DOCLINES.append(_GPU_PACKAGE_NOTE)
# pylint: disable=line-too-long # pylint: disable=line-too-long
CONSOLE_SCRIPTS = [ CONSOLE_SCRIPTS = [
'toco_from_protos = tensorflow.lite.toco.python.toco_from_protos:main', 'toco_from_protos = tensorflow.lite.toco.python.toco_from_protos:main',