Python单核与多核切换, 命令行程序设计

CLI

@click.option('--ncpu', '-j', type=int,help='How many CPUs to utilize,'
                                            ' 0 or 1 means single process,'
                                            ' -1 means utilize all CPUs, '
                                            'other number `n` menas utilize n CPUs.')

process pool

    ncpu=0
    if ncpu < 0:
        ncpu = mp.cpu_count()
    if ncpu in (0, 1):
        is_multi = False
    else:
        is_multi = True
        print('cpus count:', ncpu)
        pool = mp.Pool(ncpu)
    for i, pdbitem in enumerate(pdblist):
        args = (pdbitem, output, pose, box, i)  # your arguments
        func = modelDV.runDV  # your function
        if is_multi:
            pool.apply_async(func, args)
        else:
            func(*args)
    if is_multi:
        pool.close()
        pool.join()