Как использовать mpi4py в ноутбуке IPython?

Question

У меня есть ipcluster, работающий с 8 двигателями для ноутбука 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