可导入的`多处理.Pool`函数 [英] Importable `multiprocessing.Pool` function

问题描述

这可能很简单，我只是找不到合适的问题。

如果我使用multiprocessing.Pool执行独立脚本，我知道我应该这样做：

def foo(x):
    return x**2

if __name__ == '__main__':
    with Pool(n_jobs) as p:
        p.map(foo, list_of_inputs)

但如果我想使其成为可导入函数，我假设__name__将不再是'__main__'。这样做安全吗：

def __foo(x):
    return x**2

def bar(list_of_inputs, n_jobs):
    with Pool(n_jobs) as p:
        out = p.map(__foo, list_of_inputs)
    return out

###################################################

from test.py import bar

baz = bar(mylist, 4)

我想是这样的，因为bar只会在调用进程中调用，然后只会在子进程中调用foo，但我只是想知道这通常是否是一种安全的方法。

推荐答案

我看不出您的代码有任何问题，这些代码在我的python3和python2中运行得很好(如果我不对池使用with语句)。在我看来，这是一种安全的使用multiprocessing的方式，我不明白为什么您会经历这种无限挂。您使用的是什么版本的Python？

这篇关于可导入的`多处理.Pool`函数的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！