python - 无法使用 pool.apply_async()

Question

我只是在使用 Python 的多处理模块，但是以下代码正在连续运行，但没有给出任何输出。我究竟做错了什么？我也尝试过 pool.close() 和 pool.join() 但没有效果。

这是我迄今为止尝试过的:

import multiprocessing as mp
cpu_count = mp.cpu_count()
pool = mp.Pool(processes = cpu_count)

def func(val):
    print "\nInside Function.....\n"
    return val*val

results = []
num = 1
while cpu_count >= 1:
    results.append(pool.apply_async(func, (num,)))
    cpu_count = cpu_count - 1
    num = num + 1

output = [p.get() for p in results]
print output

Answer 1

用以下内容包装您的代码

if __name__ == '__main__':
    ...

如果您在 Windows 下工作，由于多处理实现，这是必要的。

以下代码按照您的预期打印 [1, 4, 9, 16]:

import multiprocessing as mp

def func(val):
    print "\nInside Function.....\n"
    return val*val

if __name__ == '__main__':
    cpu_count = mp.cpu_count()
    pool = mp.Pool(processes = cpu_count)

    results = []
    num = 1
    while cpu_count >= 1:
        results.append(pool.apply_async(func, (num,)))
        cpu_count = cpu_count - 1
        num = num + 1

    output = [p.get() for p in results]
    print output

    pool.close()
    pool.join()

在程序退出之前注意池关闭 - 仅终止程序并不能保证池关闭，它的进程可以保持运行。