OpenMDAO SimpleGADriver未并行运行问题求助

核心排查与修复步骤 #

• 必须通过MPI启动脚本：Windows环境下OpenMDAO的并行计算依赖MPI多进程启动，直接用python your_script.py只会运行单进程。正确启动命令示例：

  mpiexec -n 4 python your_script.py
  

  其中-n 4指定并行进程数，建议进程数不超过CPU核心数，且与种群规模匹配（比如种群大小设为进程数的整数倍）。

• 验证MPI环境有效性：conda安装的MPI可能存在路径配置问题，先确认mpi4py能正常初始化：
  在脚本开头添加代码：

  from mpi4py import MPI
  comm = MPI.COMM_WORLD
  print(f"Rank {comm.rank} of {comm.size} is running")
  

  用mpiexec启动后若仅输出rank0，说明MPI环境异常，可卸载重装：

  conda remove mpi mpi4py
  conda install -c conda-forge mpi mpi4py
  

• 检查驱动选项设置顺序：确保run_parallel=True在驱动实例化后优先设置，避免被后续配置覆盖：

  from openmdao.drivers.simple_ga_driver import SimpleGADriver
  prob.driver = SimpleGADriver()
  prob.driver.options['run_parallel'] = True
  prob.driver.options['debug_print'] = ['objs']
  

• 排查Windows系统限制：Windows防火墙可能阻止MPI进程间通信，可临时关闭防火墙测试，或给python.exe添加防火墙允许规则。

• 匹配种群规模与进程数：若种群规模远小于进程数，并行效果会不明显。建议设置种群大小为进程数的2-3倍，比如4进程对应8-12的种群规模。

额外验证手段 #

• 将time.sleep()放在Branin组件的compute方法内，确保每个个体计算有明显延迟，便于观察并行提速效果。
• 在compute方法中打印当前进程rank与计算变量值，确认多进程均在执行计算任务。

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