ipython - 如何在 IPython 笔记本中使用 mpi4py？

Question

我有一个为 Ipython 笔记本运行 8 个引擎的 ipcluster。所以以下内容：

from IPython import parallel
clients = parallel.Client()
clients.block = True  # use synchronous computations
print clients.ids

给[0, 1, 2, 3, 4, 5, 6, 7]. 但是，我无法让 mpi4py 看到这些引擎：

from mpi4py import MPI
print MPI.COMM_WORLD.size

给1. 我想，我缺少一些基本的东西，所以我很感激任何帮助。

Answer 1

事实证明，它需要一些魔法。单元格的第一行应该是：

%%px
from mpi4py import MPI
print MPI.COMM_WORLD.size