sudo apt install libopenmpi-dev pip install mpi4py
sudo apt install libopenmpi-dev
pip install mpi4py