I had a similar issue and solved it using:

sudo apt purge --autoremove libopenmpi-dev libopenmpi3 mpich openmpi-bin openmpi-common
sudo pip uninstall mpi4py

Followed by:

sudo apt install libopenmpi-dev libopenmpi3 mpich openmpi-bin openmpi-common
sudo pip install mpi4py