diff --git a/setup.py b/setup.py index 98a46a2..7425e0e 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,9 @@ apt_dependencies = [ apt_get_dependencies = [ 'mpi', - 'libopenmpi-dev' + 'libopenmpi-dev', + 'openmpi-common', + 'openmpi-bin' ] apt_nccl = [