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

Question

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