Pour que mpi4py fonctionne bien il faut construire l'environement en utilisant un mpi extern à conda.
Ici vous trouvez les pas à suivre pour mettre en place l'environement. À savoir que l'ordre des commandes est important:
module purge
module load openmpi/gnu/
4.1.4.2module load conda/4.9.2
conda create -n openmpi_test_3.7 python=3.7
conda activate openmpi_test_3.7
# INSTALLER TOUS LES PAQUETS CONDA NECESSAIRES POUR VOTRE CODE AVANT D'UTILISER PIP
# par exemple: conda install timepip install mpi4py
Un fois le code lancé, avec la commande "placement --checkme" ou "placement --jobid=<JOBID>" vous pouvez verifier que
plusieurs proces sont deployés et que chaque proces turne sur un coeur different.
Le profiler Map montre comme "Interpreter" toute la region parallelisé.
Ici vous trouvez un exemple (test_mpi4py.tar) de code python pour verifier le fonctionnement de l'environement.
Fichiers attachés
test_mpi4py.tar10 Ko