python - 我们如何在与 joblib 的并行执行中使用 tqdm？

Question

我想并行运行一个函数，并等到所有并行节点都完成，使用 joblib。就像在示例中一样:

from math import sqrt
from joblib import Parallel, delayed
Parallel(n_jobs=2)(delayed(sqrt)(i ** 2) for i in range(10))

但是，我希望像 tqdm 一样在单个进度条中看到执行，显示已完成的作业数。

你会怎么做？

Answer 1

只需将 range(10) 放入 tqdm(...) !这对你来说可能看起来太好了，但它确实有效(在我的机器上):

from math import sqrt
from joblib import Parallel, delayed  
from tqdm import tqdm  
result = Parallel(n_jobs=2)(delayed(sqrt)(i ** 2) for i in tqdm(range(100000)))