Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -0,0 +1,173 @@
easyblock = 'PythonBundle'

name = 'TensorFlow'
version = '2.3.0'
versionsuffix = '-Python-%(pyver)s'

homepage = 'https://www.tensorflow.org/'
description = "An open-source software library for Machine Intelligence"

toolchain = {'name': 'foss', 'version': '2019b'}
toolchainopts = {'usempi': True, 'pic': True}

builddependencies = [
('Bazel', '3.4.1'),
('protobuf', '3.10.0'),
# git 2.x required, see also https://github.com/tensorflow/tensorflow/issues/29053
('git', '2.23.0', '-nodocs'),
('pybind11', '2.4.3', versionsuffix),
]
dependencies = [
('Python', '3.7.4'),
('h5py', '2.10.0', versionsuffix),
('cURL', '7.66.0'),
('double-conversion', '3.1.4'),
('flatbuffers', '1.12.0'),
('giflib', '5.2.1'),
('hwloc', '1.11.12'),
('ICU', '64.2'),
('JsonCpp', '1.9.3'),
('libjpeg-turbo', '2.0.3'),
('LMDB', '0.9.24'),
('NASM', '2.14.02'),
('nsync', '1.24.0'),
('SQLite', '3.29.0'),
('PCRE', '8.43'),
('protobuf-python', '3.10.0', versionsuffix),
('libpng', '1.6.37'),
('snappy', '1.1.7'),
('SWIG', '4.0.1'),
('zlib', '1.2.11'),
# TF 2.3 requires SciPy 1.4.1 due to potential crashes with other versions
# See https://github.com/tensorflow/tensorflow/commit/54daf3c5700897a6062313983933ca28e92c410d
# Add at bottom so it will be loaded after any dependency loading the SciPy-Bundle
('scipy', '1.4.1', versionsuffix),
]

exts_default_options = {
'source_urls': [PYPI_SOURCE],
'sanity_pip_check': True,
}
use_pip = True

# Dependencies created and updated using findPythonDeps.sh:
# https://gist.github.com/Flamefire/49426e502cd8983757bd01a08a10ae0d
exts_list = [
('Markdown', '3.2.2', {
'checksums': ['1fafe3f1ecabfb514a5285fca634a53c1b32a81cb0feb154264d55bf2ff22c17'],
}),
('pyasn1-modules', '0.2.8', {
'checksums': ['905f84c712230b2c592c19470d3ca8d552de726050d1d1716282a1f6146be65e'],
}),
('rsa', '4.6', {
'checksums': ['109ea5a66744dd859bf16fe904b8d8b627adafb9408753161e766a92e7d681fa'],
}),
('cachetools', '4.1.1', {
'checksums': ['bbaa39c3dede00175df2dc2b03d0cf18dd2d32a7de7beb68072d13043c9edb20'],
}),
('google-auth', '1.20.0', {
'modulename': 'google.auth',
'checksums': ['c6e9735a2ee829a75b546702e460489db5cc35567a27fabd70b7c459f11efd58'],
}),
('oauthlib', '3.1.0', {
'checksums': ['bee41cc35fcca6e988463cacc3bcb8a96224f470ca547e697b604cc697b2f889'],
}),
('requests-oauthlib', '1.3.0', {
'checksums': ['b4261601a71fd721a8bd6d7aa1cc1d6a8a93b4a9f5e96626f8e4d91e8beeaa6a'],
}),
('google-auth-oauthlib', '0.4.1', {
'checksums': ['88d2cd115e3391eb85e1243ac6902e76e77c5fe438b7276b297fbe68015458dd'],
}),
('Werkzeug', '1.0.1', {
'checksums': ['6c80b1e5ad3665290ea39320b91e1be1e0d5f60652b964a3070216de83d2e47c'],
}),
('absl-py', '0.9.0', {
'modulename': 'absl',
'checksums': ['75e737d6ce7723d9ff9b7aa1ba3233c34be62ef18d5859e706b8fdc828989830'],
}),
('astunparse', '1.6.3', {
'checksums': ['5ad93a8456f0d084c3456d059fd9a92cce667963232cbf763eac3bc5b7940872'],
}),
('grpcio', '1.30.0', {
'modulename': 'grpc',
'checksums': ['e8f2f5d16e0164c415f1b31a8d9a81f2e4645a43d1b261375d6bab7b0adf511f'],
}),
('tensorboard-plugin-wit', '1.7.0', {
'source_tmpl': 'tensorboard_plugin_wit-%(version)s-py3-none-any.whl',
'unpack_sources': False,
'checksums': ['ee775f04821185c90d9a0e9c56970ee43d7c41403beb6629385b39517129685b'],
}),
('tensorboard', version, {
'source_tmpl': 'tensorboard-%(version)s-py3-none-any.whl',
'unpack_sources': False,
'checksums': ['d34609ed83ff01dd5b49ef81031cfc9c166bba0dabd60197024f14df5e8eae5e'],
}),
('google-pasta', '0.2.0', {
'modulename': 'pasta',
'checksums': ['c9f2c8dfc8f96d0d5808299920721be30c9eec37f2389f28904f454565c8a16e'],
}),
('termcolor', '1.1.0', {
'checksums': ['1d6d69ce66211143803fbc56652b41d73b4a400a2891d7bf7a1cdf4c02de613b'],
}),
('tensorflow-estimator', version, {
'source_tmpl': 'tensorflow_estimator-%(version)s-py2.py3-none-any.whl',
'unpack_sources': False,
'checksums': ['b75e034300ccb169403cf2695adf3368da68863aeb0c14c3760064c713d5c486'],
}),
('astor', '0.8.0', {
'checksums': ['37a6eed8b371f1228db08234ed7f6cfdc7817a3ed3824797e20cbb11dc2a7862'],
}),
('Keras-Applications', '1.0.8', {
'source_tmpl': 'Keras_Applications-%(version)s.tar.gz',
'checksums': ['5579f9a12bcde9748f4a12233925a59b93b73ae6947409ff34aa2ba258189fe5'],
}),
('gast', '0.3.3', {
'checksums': ['b881ef288a49aa81440d2c5eb8aeefd4c2bb8993d5f50edae7413a85bfdb3b57'],
}),
('opt-einsum', '3.3.0', {
'source_tmpl': 'opt_einsum-%(version)s.tar.gz',
'checksums': ['59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549'],
}),
('wrapt', '1.12.1', {
'checksums': ['b62ffa81fb85f4332a4f609cab4ac40709470da05643a082ec1eb88e6d9b97d7'],
}),
('Keras-Preprocessing', '1.1.2', {
'source_tmpl': 'Keras_Preprocessing-%(version)s.tar.gz',
'checksums': ['add82567c50c8bc648c14195bf544a5ce7c1f76761536956c3d2978970179ef3'],
}),
(name, version, {
'patches': [
'TensorFlow-2.1.0_fix-cuda-build.patch',
'TensorFlow-2.1.0_fix-system-nasm.patch',
'TensorFlow-2.3.0_fix-json-include-style.patch',
'TensorFlow-2.3.0_fix-system_absl_py.patch',
'TensorFlow-2.3.0_fix-system-protobuf.patch',
'TensorFlow-2.3.0_fix-system-protobuf2.patch',
'TensorFlow-2.3.0_fix-protoc-build.patch',
'TensorFlow-2.3.0_revert-tools-to-exectools-renaming.patch',
],
'source_tmpl': 'v%(version)s.tar.gz',
'source_urls': ['https://github.com/tensorflow/tensorflow/archive/'],
'test_script': 'TensorFlow-2.x_mnist-test.py',
'checksums': [
'2595a5c401521f20a2734c4e5d54120996f8391f00bb62a57267d930bce95350', # v2.3.0.tar.gz
'78c20aeaa7784b8ceb46238a81e8c2461137d28e0b576deeba8357d23fbe1f5a', # TensorFlow-2.1.0_fix-cuda-build.patch
# TensorFlow-2.1.0_fix-system-nasm.patch
'6671e40d60edaf1e57b1861aa3b2178d48f9b7dfb5b5c0d44db541116f848f2a',
# TensorFlow-2.3.0_fix-json-include-style.patch
'e04f5d9aa4c1222393b2cad4b84cd63128ef56eb5cc732d2afbc5d98dfa596a4',
# TensorFlow-2.3.0_fix-system_absl_py.patch
'04e53b637ca64c4804a4a8d636d75e3c09257034b143b70a33bfee09e6ab4b10',
# TensorFlow-2.3.0_fix-system-protobuf.patch
'4407c3f126eb9d3c1658ef8d6ab3a26e7ed2a40ac869dd194d8a1282f6432c1d',
# TensorFlow-2.3.0_fix-system-protobuf2.patch
'74718e007acb7f5f016848218421895a33e9e894620154482deb75408225577d',
# TensorFlow-2.3.0_fix-protoc-build.patch
'78a8390ea5e06dfcfb25f6434721abbe678e37c4ad7f052aad766c4df399bd53',
# TensorFlow-2.3.0_revert-tools-to-exectools-renaming.patch
'1a72f072a764bf66733445ce027c0bad7de7c88092363c0e2a7b91a422608d56',
],
}),
]

moduleclass = 'lib'
Original file line number Diff line number Diff line change
@@ -0,0 +1,175 @@
easyblock = 'PythonBundle'

name = 'TensorFlow'
version = '2.3.0'
versionsuffix = '-Python-%(pyver)s'

homepage = 'https://www.tensorflow.org/'
description = "An open-source software library for Machine Intelligence"

toolchain = {'name': 'fosscuda', 'version': '2019b'}
toolchainopts = {'usempi': True, 'pic': True}

builddependencies = [
('Bazel', '3.4.1'),
('protobuf', '3.10.0'),
# git 2.x required, see also https://github.com/tensorflow/tensorflow/issues/29053
('git', '2.23.0', '-nodocs'),
('pybind11', '2.4.3', versionsuffix),
]
dependencies = [
('cuDNN', '7.6.4.38'),
('NCCL', '2.4.8'),
('Python', '3.7.4'),
('h5py', '2.10.0', versionsuffix),
('cURL', '7.66.0'),
('double-conversion', '3.1.4'),
('flatbuffers', '1.12.0'),
('giflib', '5.2.1'),
('hwloc', '1.11.12'),
('ICU', '64.2'),
('JsonCpp', '1.9.3'),
('libjpeg-turbo', '2.0.3'),
('LMDB', '0.9.24'),
('NASM', '2.14.02'),
('nsync', '1.24.0'),
('SQLite', '3.29.0'),
('PCRE', '8.43'),
('protobuf-python', '3.10.0', versionsuffix),
('libpng', '1.6.37'),
('snappy', '1.1.7'),
('SWIG', '4.0.1'),
('zlib', '1.2.11'),
# TF 2.3 requires SciPy 1.4.1 due to potential crashes with other versions
# See https://github.com/tensorflow/tensorflow/commit/54daf3c5700897a6062313983933ca28e92c410d
# Add at bottom so it will be loaded after any dependency loading the SciPy-Bundle
('scipy', '1.4.1', versionsuffix),
]

exts_default_options = {
'source_urls': [PYPI_SOURCE],
'sanity_pip_check': True,
}
use_pip = True

# Dependencies created and updated using findPythonDeps.sh:
# https://gist.github.com/Flamefire/49426e502cd8983757bd01a08a10ae0d
exts_list = [
('Markdown', '3.2.2', {
'checksums': ['1fafe3f1ecabfb514a5285fca634a53c1b32a81cb0feb154264d55bf2ff22c17'],
}),
('pyasn1-modules', '0.2.8', {
'checksums': ['905f84c712230b2c592c19470d3ca8d552de726050d1d1716282a1f6146be65e'],
}),
('rsa', '4.6', {
'checksums': ['109ea5a66744dd859bf16fe904b8d8b627adafb9408753161e766a92e7d681fa'],
}),
('cachetools', '4.1.1', {
'checksums': ['bbaa39c3dede00175df2dc2b03d0cf18dd2d32a7de7beb68072d13043c9edb20'],
}),
('google-auth', '1.20.0', {
'modulename': 'google.auth',
'checksums': ['c6e9735a2ee829a75b546702e460489db5cc35567a27fabd70b7c459f11efd58'],
}),
('oauthlib', '3.1.0', {
'checksums': ['bee41cc35fcca6e988463cacc3bcb8a96224f470ca547e697b604cc697b2f889'],
}),
('requests-oauthlib', '1.3.0', {
'checksums': ['b4261601a71fd721a8bd6d7aa1cc1d6a8a93b4a9f5e96626f8e4d91e8beeaa6a'],
}),
('google-auth-oauthlib', '0.4.1', {
'checksums': ['88d2cd115e3391eb85e1243ac6902e76e77c5fe438b7276b297fbe68015458dd'],
}),
('Werkzeug', '1.0.1', {
'checksums': ['6c80b1e5ad3665290ea39320b91e1be1e0d5f60652b964a3070216de83d2e47c'],
}),
('absl-py', '0.9.0', {
'modulename': 'absl',
'checksums': ['75e737d6ce7723d9ff9b7aa1ba3233c34be62ef18d5859e706b8fdc828989830'],
}),
('astunparse', '1.6.3', {
'checksums': ['5ad93a8456f0d084c3456d059fd9a92cce667963232cbf763eac3bc5b7940872'],
}),
('grpcio', '1.30.0', {
'modulename': 'grpc',
'checksums': ['e8f2f5d16e0164c415f1b31a8d9a81f2e4645a43d1b261375d6bab7b0adf511f'],
}),
('tensorboard-plugin-wit', '1.7.0', {
'source_tmpl': 'tensorboard_plugin_wit-%(version)s-py3-none-any.whl',
'unpack_sources': False,
'checksums': ['ee775f04821185c90d9a0e9c56970ee43d7c41403beb6629385b39517129685b'],
}),
('tensorboard', version, {
'source_tmpl': 'tensorboard-%(version)s-py3-none-any.whl',
'unpack_sources': False,
'checksums': ['d34609ed83ff01dd5b49ef81031cfc9c166bba0dabd60197024f14df5e8eae5e'],
}),
('google-pasta', '0.2.0', {
'modulename': 'pasta',
'checksums': ['c9f2c8dfc8f96d0d5808299920721be30c9eec37f2389f28904f454565c8a16e'],
}),
('termcolor', '1.1.0', {
'checksums': ['1d6d69ce66211143803fbc56652b41d73b4a400a2891d7bf7a1cdf4c02de613b'],
}),
('tensorflow-estimator', version, {
'source_tmpl': 'tensorflow_estimator-%(version)s-py2.py3-none-any.whl',
'unpack_sources': False,
'checksums': ['b75e034300ccb169403cf2695adf3368da68863aeb0c14c3760064c713d5c486'],
}),
('astor', '0.8.0', {
'checksums': ['37a6eed8b371f1228db08234ed7f6cfdc7817a3ed3824797e20cbb11dc2a7862'],
}),
('Keras-Applications', '1.0.8', {
'source_tmpl': 'Keras_Applications-%(version)s.tar.gz',
'checksums': ['5579f9a12bcde9748f4a12233925a59b93b73ae6947409ff34aa2ba258189fe5'],
}),
('gast', '0.3.3', {
'checksums': ['b881ef288a49aa81440d2c5eb8aeefd4c2bb8993d5f50edae7413a85bfdb3b57'],
}),
('opt-einsum', '3.3.0', {
'source_tmpl': 'opt_einsum-%(version)s.tar.gz',
'checksums': ['59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549'],
}),
('wrapt', '1.12.1', {
'checksums': ['b62ffa81fb85f4332a4f609cab4ac40709470da05643a082ec1eb88e6d9b97d7'],
}),
('Keras-Preprocessing', '1.1.2', {
'source_tmpl': 'Keras_Preprocessing-%(version)s.tar.gz',
'checksums': ['add82567c50c8bc648c14195bf544a5ce7c1f76761536956c3d2978970179ef3'],
}),
(name, version, {
'patches': [
'TensorFlow-2.1.0_fix-cuda-build.patch',
'TensorFlow-2.1.0_fix-system-nasm.patch',
'TensorFlow-2.3.0_fix-json-include-style.patch',
'TensorFlow-2.3.0_fix-system_absl_py.patch',
'TensorFlow-2.3.0_fix-system-protobuf.patch',
'TensorFlow-2.3.0_fix-system-protobuf2.patch',
'TensorFlow-2.3.0_fix-protoc-build.patch',
'TensorFlow-2.3.0_revert-tools-to-exectools-renaming.patch',
],
'source_tmpl': 'v%(version)s.tar.gz',
'source_urls': ['https://github.com/tensorflow/tensorflow/archive/'],
'test_script': 'TensorFlow-2.x_mnist-test.py',
'checksums': [
'2595a5c401521f20a2734c4e5d54120996f8391f00bb62a57267d930bce95350', # v2.3.0.tar.gz
'78c20aeaa7784b8ceb46238a81e8c2461137d28e0b576deeba8357d23fbe1f5a', # TensorFlow-2.1.0_fix-cuda-build.patch
# TensorFlow-2.1.0_fix-system-nasm.patch
'6671e40d60edaf1e57b1861aa3b2178d48f9b7dfb5b5c0d44db541116f848f2a',
# TensorFlow-2.3.0_fix-json-include-style.patch
'e04f5d9aa4c1222393b2cad4b84cd63128ef56eb5cc732d2afbc5d98dfa596a4',
# TensorFlow-2.3.0_fix-system_absl_py.patch
'04e53b637ca64c4804a4a8d636d75e3c09257034b143b70a33bfee09e6ab4b10',
# TensorFlow-2.3.0_fix-system-protobuf.patch
'4407c3f126eb9d3c1658ef8d6ab3a26e7ed2a40ac869dd194d8a1282f6432c1d',
# TensorFlow-2.3.0_fix-system-protobuf2.patch
'74718e007acb7f5f016848218421895a33e9e894620154482deb75408225577d',
# TensorFlow-2.3.0_fix-protoc-build.patch
'78a8390ea5e06dfcfb25f6434721abbe678e37c4ad7f052aad766c4df399bd53',
# TensorFlow-2.3.0_revert-tools-to-exectools-renaming.patch
'1a72f072a764bf66733445ce027c0bad7de7c88092363c0e2a7b91a422608d56',
],
}),
]

moduleclass = 'lib'
Loading