mirror of
https://code.it4i.cz/sccs/easyconfigs-it4i.git
synced 2025-04-11 01:12:11 +01:00
58 lines
1.5 KiB
Plaintext
58 lines
1.5 KiB
Plaintext
easyblock = 'CMakeMake'
|
|
|
|
name = 'Caffe'
|
|
version = '1.0'
|
|
|
|
cudaversion = '8.0.44'
|
|
versionsuffix = '-CUDA-%s-Python-%%(pyver)s' % cudaversion
|
|
|
|
homepage = 'https://github.com/BVLC/caffe'
|
|
description = """
|
|
Caffe is a deep learning framework made with expression, speed,
|
|
and modularity in mind. It is developed by the Berkeley Vision
|
|
and Learning Center (BVLC) and community contributors.
|
|
"""
|
|
|
|
toolchain = {'name': 'foss', 'version': '2016a'}
|
|
toolchainopts = {'optarch': True, 'pic': True}
|
|
|
|
sources = ['%(version)s.tar.gz']
|
|
source_urls = ['https://github.com/BVLC/caffe/archive/']
|
|
|
|
builddependencies = [
|
|
('CMake', '3.8.1', '', True),
|
|
]
|
|
|
|
protobuf_ver = '3.2.0'
|
|
|
|
dependencies = [
|
|
('glog', '0.3.4', '', True),
|
|
('gflags', '2.1.2', '', True),
|
|
('OpenCV', '3.1.0', '', True),
|
|
('CUDA', cudaversion, '', True),
|
|
('cuDNN', '5.1', '-CUDA-%s' % cudaversion, True),
|
|
('LMDB', '0.9.18', '', True),
|
|
('LevelDB', '1.18', '', True),
|
|
('snappy', '1.1.3', '', True),
|
|
('protobuf', protobuf_ver),
|
|
# ('protobuf-python', protobuf_ver, '-Python-%(pyver)s'),
|
|
('HDF5', '1.8.16', '-serial'),
|
|
('Boost', '1.63.0', '-Python-%(pyver)s'),
|
|
('Python', '2.7.13', '', True),
|
|
('scikit-image', '0.12.3', '-Python-%(pyver)s', True)
|
|
]
|
|
|
|
configopts = '-DBLAS=open'
|
|
|
|
modextrapaths = {'PYTHONPATH': ['python']}
|
|
|
|
sanity_check_paths = {
|
|
'files': ['bin/caffe'] +
|
|
['include/caffe/proto/caffe.pb.h'],
|
|
'dirs': [],
|
|
}
|
|
|
|
sanity_check_commands = [('python', "-c 'import caffe'")]
|
|
|
|
moduleclass = 'devel'
|