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

Question

我有一个运行 8 个引擎的 Ipython notebook 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