Comment utiliser mpi4py dans un cahier IPython?

Question

J'ai un ipcluster fonctionnant avec 8 moteurs pour le cahier d'Ipython. Ainsi, ce qui suit:

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

donne [0, 1, 2, 3, 4, 5, 6, 7]. Mais, je ne peux pas mpi4py voir ces moteurs:

from mpi4py import MPI 
print MPI.COMM_WORLD.size 

donne 1. Je suppose qu'il me manque quelque chose de basique, alors j'apprécierais toute aide.

Answer 1

Il s'avère que cela demande un peu de magie. La première ligne de la cellule doit être:

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