tqdm进度条和多处理

时间:2018-05-03 11:40:22

标签: python multiprocessing tqdm

我试图在我的程序中添加一个进度条,但是,似乎适用于其他(在其他帖子上)的解决方案对我不起作用。

Python 3.6版。

import multiprocessing as mp
import tqdm

def f(dynamic, fix1, fix2):
    return dynamic + fix1 + fix2

N = 2

fix1 = 5
fix2= 10

dynamic = range(10)

p = mp.Pool(processes = N)

for _ in tqdm.tqdm(p.starmap(f, [(d, fix1, fix2) for d in dynamic]), total = len(dynamic)):
    pass

p.close()
p.join()

任何想法为什么多处理工作(计算完成),但没有进度条?

注意:上面的例子是假的,我的功能是不同的。

其他问题:如何正确中断多处理程序?我通常在signle线程中执行的 ctrl + C 似乎会带来一些问题。

1 个答案:

答案 0 :(得分:1)

不幸的是,tqdm无法与星图一起使用。您可以使用以下内容:

def f(args):
  arg1, arg2 = args
  ... do something with arg1, arg2 ...


for _ in tqdm.tqdm(pool.imap_unordered(f, zip(list_of_args, list_of_args2)), total=total):
    pass