enable mpi on training server