文档中的示例在 Jupiter Notebook 中不起作用

Question

我看过文档。 有一个例子

from multiprocessing import Pool

def f(x):
    return x*x

if __name__ == '__main__':
    with Pool(5) as p:
        print(p.map(f, [1, 2, 3]))

问题是：它不起作用。 我在 Jupiter Notebook 单元中运行此代码。 而这个单元不会引发任何异常。 但木星的终端确实如此。 它说： AttributeError: Can't get attribute 'f' on <module '__main__' (built-in)>

正如这里所写的，问题可能是因为我不使用__name__ == '__main__'条件。 但是我愿意。

我从文档中复制并粘贴了示例，但它不起作用。 我该怎么办？

Answer 1

我怀疑你在 Windows 上运行。 如果是这样，这是一个已知问题。 请参阅这篇文章。 您需要将函数f添加到文件中，例如worker.py ：

工人.py

def f(x):
    return x*x

然后你的 jupyter notebook 代码变成了：

from multiprocessing import Pool
import worker


if __name__ == '__main__':
    with Pool(5) as p:
        print(p.map(worker.f, [1, 2, 3]))