python-2.7 - 多处理池卡在 jupyter notebook 中

Question

我有一个非常简单的脚本如下:

import multiprocessing as multi

def call_other_thing_with_multi():
    P = multi.Pool(3)
    P.map(other_thing, range(0,5))
    P.join()


def other_thing(arg):
    print(arg)
    return arg**2.

call_other_thing_with_multi()

当我调用它时，我的代码永远挂起。这是在带有 python 2.7 的 Windows 上。

感谢您的指导!

Answer 1

根据 documentation ，您需要在 join() 之前调用 close():

import multiprocessing as multi

def call_other_thing_with_multi():
    P = multi.Pool(3)
    P.map(other_thing, range(0,5))
    P.close() # <-- calling close before P.join()
    P.join()
    print('END')

def other_thing(arg):
    print(arg)
    return arg**2.

call_other_thing_with_multi()

打印:

0
1
2
3
4
END

编辑:最好使用上下文管理器，不要忘记调用 close():

def call_other_thing_with_multi():
    with multi.Pool(3) as P:
        P.map(other_thing, range(0,5))
    print('END')