Commit d082d588 authored by Sabry Razick's avatar Sabry Razick
Browse files

Version tested on 1 GPU on SAGA

parent 7e024be3
easyblock = 'PythonBundle'
name = 'TensorFlow'
version = '1.15.3c'
version = '1.15.3'
versionsuffix = '-Python-%(pyver)s'
homepage = 'https://www.tensorflow.org/'
......@@ -19,13 +19,29 @@ builddependencies = [
dependencies = [
('Python', '3.7.2','',('GCCcore', '8.2.0')),
('NLPL-SciPy-bundle', '2019.10', versionsuffix),
('NLPL-SciPy-bundle', '2019.10', versionsuffix, ('gomkl', '2019a')),
('NLPL-TensorDependencies', '1.0.0', versionsuffix),
('h5py', '2.10.0', 'versionsuffix',('gomkl', '2019a')),
('h5py', '2.10.0', versionsuffix,('gomkl', '2019a')),
('cuDNN', '7.6.4.38'),
('NCCL', '2.4.8'),
]
#Tesla T4 7.5
#Tesla V100 7.0
#Tesla P100 6.0
#Tesla P40 6.1
#Tesla P4 6.1
#Tesla M60 5.2
#Tesla M40 5.2
#Tesla K80 3.7
#Tesla K40 3.5
#Tesla K20 3.5
#Tesla K10 3.0
#cuda_compute_capabilities = ['3.0', '3.2', '3.5', '3.7', '5.0', '6.0', '7.0']
cuda_compute_capabilities = ['6.0'] # Saga.sigma2.no, Tesla P100-PCIE, found
# using nvidia-smi
exts_default_options = {
'source_urls': [PYPI_SOURCE],
'sanity_pip_check': True,
......@@ -33,89 +49,29 @@ exts_default_options = {
use_pip = True
exts_list = [
('Keras-Preprocessing', '1.1.0', {
'modulename': 'keras_preprocessing',
'source_tmpl': 'Keras_Preprocessing-%(version)s.tar.gz',
'checksums': ['5a8debe01d840de93d49e05ccf1c9b81ae30e210d34dacbcc47aeb3049b528e5'],
}),
('absl-py', '0.9.0', {
'modulename': 'absl',
'checksums': ['75e737d6ce7723d9ff9b7aa1ba3233c34be62ef18d5859e706b8fdc828989830'],
}),
('grpcio', '1.27.2', {
'modulename': 'grpc',
'checksums': ['5ae532b93cf9ce5a2a549b74a2c35e3b690b171ece9358519b3039c7b84c887e'],
}),
('opt-einsum', '3.2.0', {
'source_tmpl': 'opt_einsum-%(version)s.tar.gz',
'checksums': ['738b0a1db1d3084d360081bb64d826f9db06d2df7cc0bf8e2c9356028da1fa31'],
}),
('gast', '0.2.2', {
'checksums': ['fe939df4583692f0512161ec1c880e0a10e71e6a232da045ab8edd3756fbadf0'],
}),
('tensorflow-estimator', '1.15.1', {
'source_tmpl': 'tensorflow_estimator-%(version)s-py2.py3-none-any.whl',
'unpack_sources': False,
'checksums': ['8853bfb7c3c96fbdc80b3d66c37a10af6ccbcd235dc87474764270c02a0f86b9'],
}),
('google-pasta', '0.1.8', {
'modulename': 'pasta',
'checksums': ['713813a9f7d6589e5defdaf21e80e4392eb124662f8bd829acd51a4f8735c0cb'],
}),
('wrapt', '1.12.0', {
'checksums': ['0ec40d9fd4ec9f9e3ff9bdd12dbd3535f4085949f4db93025089d7a673ea94e8'],
}),
('Werkzeug', '1.0.0', {
'checksums': ['169ba8a33788476292d04186ab33b01d6add475033dfc07215e6d219cc077096'],
}),
('Markdown', '3.2.1', {
'checksums': ['90fee683eeabe1a92e149f7ba74e5ccdc81cd397bd6c516d93a8da0ef90b6902'],
}),
('protobuf', '3.11.3', {
'modulename': 'google.protobuf',
'checksums': ['c77c974d1dadf246d789f6dad1c24426137c9091e930dbf50e0a29c1fcf00b1f'],
}),
('tensorboard', '1.15.0', {
'source_tmpl': 'tensorboard-%(version)s-py3-none-any.whl',
'unpack_sources': False,
'checksums': ['4cad2c65f6011e51609b463014c014fd7c6ddd9c1263af1d4f18dd97ed88c2bc'],
}),
('Keras-Applications', '1.0.8', {
'modulename': 'keras_applications',
'source_tmpl': 'Keras_Applications-%(version)s.tar.gz',
'checksums': ['5579f9a12bcde9748f4a12233925a59b93b73ae6947409ff34aa2ba258189fe5'],
}),
('termcolor', '1.1.0', {
'checksums': ['1d6d69ce66211143803fbc56652b41d73b4a400a2891d7bf7a1cdf4c02de613b'],
}),
('astor', '0.8.1', {
'checksums': ['6a6effda93f4e1ce9f618779b2dd1d9d84f1e32812c23a29b3fff6fd7f63fa5e'],
}),
(name, version, {
'patches': [
'TensorFlow-1.14.0_swig-env.patch',
'TensorFlow-1.15.0_remove-msse-hardcoding.patch',
'TensorFlow-1.15.0_lrt-flag.patch',
'TensorFlow-1.14.0_remove_usrbin_from_linker_bin_path_flag.patch',
'TensorFlow-1.14.0_fix-mpi-undeclared-inclusion.patch',
'TensorFlow-1.14.0_fix-cuda-build.patch',
'TensorFlow-1.14.0_fix-cuda-mpi.patch',
'TensorFlow-1.15.3_protobuf.patch',
'TensorFlow-1.15.3_bazel-conf.patch',
'TensorFlow-1.15.3_mpi_collectives_BUILD.patch',
'TensorFlow-1.15.3_core-build.patch',
'TensorFlow-1.15.3_cuda_conf.patch',
],
'source_tmpl': 'v%(version)s.tar.gz',
'source_urls': ['https://github.com/tensorflow/tensorflow/archive/'],
'checksums': [
'd95d75d26a298211b5e802842e87fda5b8b14f6ad83719377b391e5fb71b8746', # v1.15.2.tar.gz
'b83cce6b91c7d19b8b320158ffc50fb4b2de454f5ac191c58d704234a1bf9005', # TensorFlow-1.14.0_swig-env.patch
# TensorFlow-1.15.0_remove-msse-hardcoding.patch
'59e408f1cf5a97d90f33861123d55eb1332b8fef1d5d8fdfdea413c4b481ee56',
'b0fd4c7902be45bba18bd04192800852b140a9cf312a44ac1efb7ee653d3d886', # TensorFlow-1.15.0_lrt-flag.patch
# TensorFlow-1.14.0_remove_usrbin_from_linker_bin_path_flag.patch
'bfa2701495dddfbbf4fbec808ea3c982b50043a16d28a8d1283f44df0b8aae7d',
# TensorFlow-1.14.0_fix-mpi-undeclared-inclusion.patch
'09e5fdce89588074b3c2abb6a4705b1f141b43395c960660320cf1cb79cd1ef4',
# TensorFlow-1.14.0_fix-cuda-build.patch
'92afeee76520a63dc638c2cd4adaf2e7429411068ea4be96718255f67f113e43',
'fbc43da0de293bf7f42810ee439543b47cc7e19107b9ad7b6db95cec21eb8b33', # TensorFlow-1.14.0_fix-cuda-mpi.patch
'9ab1d92e58eb813922b040acc7622b32d73c2d8d971fe6491a06f9df4c778151', # v1.15.3.tar.gz
],
}),
]
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment