diff --git a/requirements.txt b/requirements.txt
index a5ec293..57039c2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,13 +1,13 @@
 hyperopt>=0.1.2
 matplotlib>=3.0.3
-mpi4py==3.0.2
+mpi4py[MPI]==3.0.2
 numpy>=1.16.2
 optuna>=0.9.0
 Optunity>=1.1.1
 pandas>=0.24.2
 pytest>=4.3.1
 scikit-learn>=0.20.3
 scipy>=1.2.1
 visdom>=0.1.8.8
 xmlrunner>=1.7.7
 Sphinx>=1.8.3
\ No newline at end of file