Pop OS下MPICH运行mpi4py程序异常及hwloc警告求助

核心问题分析 #

你遇到的0 of 1重复输出，说明每个Python进程都初始化了独立的MPI环境，没有形成真正的并行通信域。结合你的MPICH编译参数（--prefix=NONE）和环境配置，问题大概率出在MPICH安装不规范或mpi4py与MPICH库不匹配上。

分步解决 #

1. 验证mpi4py绑定的MPI库 #

先确认mpi4py是否正确关联到MPICH：

python -c "from mpi4py import MPI; print(MPI.get_vendor())"

如果输出不是('MPICH', (4, 2, 1))（匹配你的MPICH版本），说明mpi4py链接了其他MPI库（如OpenMPI），需要重新绑定。

2. 重新规范安装MPICH #

你之前编译时使用--prefix=NONE，导致MPICH没有安装到系统可识别的标准路径，这会引发库查找问题。重新编译安装：

# 进入MPICH源码目录
cd ~/Downloads/mpich-4.2.1
# 清理之前的编译文件
make clean
# 配置安装路径（推荐/usr/local/mpich）
./configure --prefix=/usr/local/mpich
# 并行编译（nproc为CPU核心数）
make -j$(nproc)
# 安装到系统路径
sudo make install

3. 配置MPICH环境变量 #

将MPICH的bin和lib路径加入环境变量，编辑~/.bashrc（若使用zsh则编辑~/.zshrc）：

echo 'export PATH=/usr/local/mpich/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/mpich/lib:$LD_LIBRARY_PATH' >> ~/.bashrc
# 生效环境变量
source ~/.bashrc

验证配置：

which mpiexec
# 应输出/usr/local/mpich/bin/mpiexec
mpiexec --version
# 应显示你的MPICH 4.2.1版本

4. 重新安装mpi4py并绑定MPICH #

卸载现有mpi4py，用MPICH的编译器重新安装：

pip uninstall -y mpi4py
# 指定MPICH的mpicc编译器路径
MPICC=/usr/local/mpich/bin/mpicc pip install mpi4py

5. 测试并行运行 #

再次执行你的测试程序：

mpiexec -n 4 python hello.py

正常输出应为：

0 of 4
1 of 4
2 of 4
3 of 4

关于hwloc警告的处理 #

你看到的hwloc/linux: Ignoring PCI device with non-16bit domain警告，是hwloc库对PCI设备的识别兼容性问题，不影响MPI并行功能。如果一定要消除警告：

• 可以重新编译系统的hwloc库，添加--enable-32bits-pci-domain参数（注意：这会破坏ABI，非必要不推荐）
• 或者直接忽略该警告，不影响核心功能

内容的提问来源于stack exchange，提问作者ray_lv