如何让使用mpi4py编写的Python MPI脚本正常结束执行？

你的MPI程序无法自动结束的核心问题是只有rank 0进程执行了终止逻辑，其他worker进程没有正常退出，导致整个MPI通信域内还有活跃进程，程序无法完全终止。

问题分析 #

原代码中，rank 0进程执行完任务后调用了MPI.Finalize()并exit(0)，但其他rank的进程在完成fun_worker(rank)后，没有任何终止或同步逻辑，会一直处于存活状态。MPI运行时要求所有进程都正常结束才会终止整个任务，因此你必须手动取消。

另外，mpi4py通常不需要手动调用MPI.Finalize()——Python解释器退出时会自动处理，但前提是所有进程都能正常走到程序结束的位置。

修复后的代码示例 #

from mpi4py import MPI

def fun():
    # 你的fun函数实现
    return "处理结果"

def fun_worker(rank):
    # 你的worker进程逻辑
    pass

if __name__ == "__main__":
    comm = MPI.COMM_WORLD
    size = comm.Get_size()
    rank = comm.Get_rank()
    
    if rank == 0:
        print("MPI WORLD SIZE: ", size)
        res = fun()
        print(res)
    else:
        fun_worker(rank)
    
    # 所有进程统一执行Finalize（可选，mpi4py会自动处理，但显式调用更清晰）
    MPI.Finalize()

额外检查点 #

如果修复后还是无法自动结束，需要排查fun_worker的逻辑：

• 是否存在阻塞的MPI通信操作（比如comm.recv()没有对应的发送方，导致进程一直等待）
• 是否有无限循环或未处理的异常导致worker进程无法正常返回

内容的提问来源于stack exchange，提问作者Pepe