0df70e217f
* A collection of tensorflow fixes and updates * tensorflow 2.3.1 requires the workaround for external protobuf as well * Update tensorflow-estimator to 2.3.0 * Update tensorboard to 2.3.0 * Update tensorboard-plugin-wit to use actual releases * Patch that potentially fixes #16073 * add myself to maintainer list
82 lines
3.5 KiB
Python
82 lines
3.5 KiB
Python
# Copyright 2013-2020 Lawrence Livermore National Security, LLC and other
|
|
# Spack Project Developers. See the top-level COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
from spack import *
|
|
|
|
|
|
class PyTensorflowEstimator(Package):
|
|
"""TensorFlow Estimator is a high-level TensorFlow API that greatly
|
|
simplifies machine learning programming."""
|
|
|
|
homepage = "https://github.com/tensorflow/estimator"
|
|
url = "https://github.com/tensorflow/estimator/archive/v2.2.0.tar.gz"
|
|
|
|
maintainers = ['aweits']
|
|
|
|
version('2.3.0', sha256='75403e7de7e8ec30ec0781ede56ed84cbe5e90daad64a9c242cd489c8fe63a17')
|
|
version('2.2.0', sha256='2d68cb6e6442e7dcbfa2e092aa25bdcb0eda420536a829b85d732854a4c85d46')
|
|
version('2.1', sha256='1d74c8181b981748976fa33ad97d3434c3cf2b7e29a0b00861365fe8329dbc4e')
|
|
version('2.0.0', sha256='6f4bdf1ab219e1f1cba25d2af097dc820f56479f12a839853d97422fe4d8b465')
|
|
version('1.13.0', sha256='a787b150ff436636df723e507019c72a5d6486cfe506886279d380166953f12f')
|
|
|
|
extends('python')
|
|
|
|
depends_on('py-tensorflow@2.3.0:', when='@2.3.0')
|
|
depends_on('py-tensorflow@2.2.0', when='@2.2.0')
|
|
depends_on('py-tensorflow@2.1.0:2.1.999', when='@2.1')
|
|
depends_on('py-tensorflow@2.0.0:2.0.999', when='@2.0.0')
|
|
depends_on('py-tensorflow@1.13.1', when='@1.13.0')
|
|
|
|
depends_on('py-absl-py@0.7.0:', type=('build', 'run'), when='@1.12.1,1.14:')
|
|
depends_on('py-numpy@1.16.0:1.18', type=('build', 'run'), when='@1.13.2,1.15:')
|
|
depends_on('py-six@1.12.0:', type=('build', 'run'), when='@2.1:')
|
|
depends_on('py-keras-preprocessing@1.1.1:1.999', type=('build', 'run'), when='@2.3:')
|
|
|
|
depends_on('bazel@0.19.0:', type='build')
|
|
depends_on('py-funcsigs@1.0.2:', type=('build', 'run'))
|
|
|
|
def install(self, spec, prefix):
|
|
tmp_path = join_path(env.get('SPACK_TMPDIR', '/tmp/spack'),
|
|
'tf-estimator',
|
|
self.module.site_packages_dir[1:])
|
|
mkdirp(tmp_path)
|
|
env['TEST_TMPDIR'] = tmp_path
|
|
env['HOME'] = tmp_path
|
|
|
|
args = [
|
|
# Don't allow user or system .bazelrc to override build settings
|
|
'--nohome_rc',
|
|
'--nosystem_rc',
|
|
# Bazel does not work properly on NFS, switch to /tmp
|
|
'--output_user_root=' + tmp_path,
|
|
'build',
|
|
# Spack logs don't handle colored output well
|
|
'--color=no',
|
|
'--jobs={0}'.format(make_jobs),
|
|
# Enable verbose output for failures
|
|
'--verbose_failures',
|
|
# Show (formatted) subcommands being executed
|
|
'--subcommands=pretty_print',
|
|
# Ask bazel to explain what it's up to
|
|
# Needs a filename as argument
|
|
'--explain=explainlogfile.txt',
|
|
# Increase verbosity of explanation,
|
|
'--verbose_explanations',
|
|
# bazel uses system PYTHONPATH instead of spack paths
|
|
'--action_env', 'PYTHONPATH={0}'.format(env['PYTHONPATH']),
|
|
'//tensorflow_estimator/tools/pip_package:build_pip_package',
|
|
]
|
|
|
|
bazel(*args)
|
|
|
|
build_pip_package = Executable(join_path(
|
|
'bazel-bin/tensorflow_estimator/tools',
|
|
'pip_package/build_pip_package'))
|
|
buildpath = join_path(self.stage.source_path, 'spack-build')
|
|
build_pip_package('--src', buildpath)
|
|
with working_dir(buildpath):
|
|
setup_py('install', '--prefix={0}'.format(prefix),
|
|
'--single-version-externally-managed', '--root=/')
|