def main():
"""
Generate a dockerfile according to the given modules to be installed.
"""
parser = argparse.ArgumentParser(description='Composer')
parser.add_argument('path')
parser.add_argument('modules', nargs='*')
parser.add_argument('--cuda-ver')
parser.add_argument('--cudnn-ver')
parser.add_argument('--ubuntu-ver', default='18.04')
parser.add_argument(
'--deepstream', action='store_true'
) #use Nvidia DeepStream as base image, overwrite above 3 options
args = parser.parse_args()
if args.deepstream:
args.cuda_ver = '10.1'
args.cudnn_ver = '7'
args.ubuntu_ver = '18.04'
in_modules = []
versions = {}
for module in args.modules:
terms = module.split('==')
m = _import(terms[0])
in_modules.append(m)
if len(terms) > 1:
versions[m] = terms[1]
composer = Composer(in_modules, args.cuda_ver, args.cudnn_ver,
args.ubuntu_ver, versions, args.deepstream)
with open(args.path, 'w') as f:
f.write(composer.to_dockerfile())
def main(path, modules):
"""
Generate a dockerfile according to the given modules to be installed.
"""
in_modules = []
versions = {}
for module in modules:
terms = module.split('==')
m = _import(terms[0])
in_modules.append(m)
if len(terms) > 1:
versions[m] = terms[1]
composer = Composer(in_modules, versions)
with open(path, 'w') as f:
f.write(composer.to_dockerfile())
def main(path, modules, cpu_only):
"""
Generate a dockerfile according to the given modules to be installed.
"""
in_modules = []
versions = {}
for module in modules:
terms = module.split('==')
m = _import(terms[0])
in_modules.append(m)
if len(terms) > 1:
versions[m] = terms[1]
composer = Composer(in_modules, versions, cpu_only=cpu_only)
with open(path, 'w') as f:
f.write(composer.to_dockerfile())
def main():
"""
Generate a dockerfile according to the given modules to be installed.
"""
parser = argparse.ArgumentParser(description='Composer')
parser.add_argument('path')
parser.add_argument('modules', nargs='*')
parser.add_argument('--cuda-ver')
args = parser.parse_args()
in_modules = []
versions = {}
for module in args.modules:
terms = module.split('==')
m = _import(terms[0])
in_modules.append(m)
if len(terms) > 1:
versions[m] = terms[1]
composer = Composer(in_modules, args.cuda_ver, versions)
with open(args.path, 'w') as f:
f.write(composer.to_dockerfile())