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

Question

我有一个运行8个引擎的ipcluster用于Ipython笔记本。以下是：

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