Windows 上的 Python 多处理运行时错误

Question

我有一个类函数（我们称之为“alpha.py”），它使用多处理 (processes=2) 来分叉一个进程，它是我编写的 Python 包的一部分。在一个单独的 Python 脚本（我们称之为“beta.py”）中，我从这个类中实例化了一个对象，并调用了使用多处理的相应函数。最后，所有这些都包装在一个包装 Python 脚本（我们称之为“gamma.py”）中，该脚本处理许多不同的类对象和函数。

基本上：

1. 从命令行运行 ./gamma.py
2. gamma.py 使用子进程并执行 beta.py
3. beta.py 从 alpha.py 类实例化一个对象并调用使用多处理 (processes=2) 的函数

这在 Mac 或 Linux 上运行没有问题。但是，它在 Windows 机器上成为一个问题，并且错误（和文档）表明我应该在某处写这个：

if __name__ == '__main__':
    freeze_support()

This other post 也提到做同样的事情。

但是，我不知道这两行的确切位置。目前，alpha.py、beta.py 或 gamma.py 都不包含 if __name__ == '__main__': 部分。如果有人能告诉我这两行应该去哪里以及背后的理由，那就太好了。

Answer 1

其实这里不需要freeze_support()。您会得到一个RuntimeError，因为您在beta 模块的顶层创建并启动了新流程。

当在 Windows 上使用multiprocessing 创建一个新进程时，将在该进程中启动一个新的 Python 解释器，它会尝试使用应该执行的目标函数导入模块。这是您的beta 模块。现在，当您导入它时，应该执行所有顶级语句，这将导致创建并重新启动一个新进程。然后，递归地从该进程中提取另一个进程，依此类推。

这很可能不是您想要的，因此当您使用 subprocess 直接运行 beta.py 时，新进程应该初始化并只启动一次。

if __name__ == '__main__': 应该放在 beta.py 中，然后在本节中为您的新进程移动初始化和启动代码。之后，当 beta.py 被导入而不直接运行时，不会启动新进程，您也不会看到任何副作用。