Linux

关注公众号 jb51net

关闭
首页 > 网站技巧 > 服务器 > Linux > Ubuntu安装和配置MPI

在Ubuntu上安装和配置MPI环境的详细步骤

作者:一步步走

在 Ubuntu 上配置 MPI(消息传递接口)环境,常用的是 OpenMPI,它是一个高效、常用的开源 MPI 实现,以下是如何在 Ubuntu 上安装和配置 MPI 环境的详细步骤,需要的朋友可以参考下

1. 安装 OpenMPI

首先,确保系统是最新的,并安装 OpenMPI。

sudo apt update
sudo apt install openmpi-bin openmpi-common libopenmpi-dev
mpirun --version
mpirun (Open MPI) 4.1.1 Report bugs to http://www.open-mpi.org/community/help/

2. 配置 OpenMPI 环境

export PATH=/usr/lib/openmpi/bin:$PATH 
export LD_LIBRARY_PATH=/usr/lib/openmpi/lib:$LD_LIBRARY_PATH
source ~/.bashrc

3. 编写 MPI 程序

接下来,我们来编写一个简单的 MPI 程序,以确保安装和配置正确。

// hello_mpi.c 
#include <mpi.h> 
#include <stdio.h> 
int main(int argc, char** argv) 
{
    MPI_Init(&argc, &argv); // 初始化MPI环境 
    int world_rank; 
    MPI_Comm_rank(MPI_COMM_WORLD, &world_rank); // 获取当前进程的Rank 
    int world_size; 
    MPI_Comm_size(MPI_COMM_WORLD, &world_size); // 获取总进程数 
    printf("Hello world from processor %d out of %d processors\n", world_rank, world_size);
    MPI_Finalize(); // 结束MPI环境 
    return 0; 
}
mpicc hello_mpi.c -o hello_mpi
mpirun -np 4 ./hello_mpi

该命令使用 4 个进程运行程序。你应该会看到类似以下的输出:

Hello world from processor 0 out of 4 processors 
Hello world from processor 1 out of 4 processors 
Hello world from processor 2 out of 4 processors 
Hello world from processor 3 out of 4 processors

4. 运行多节点的 MPI 程序(可选)

如果你有多台机器并希望在集群上运行 MPI 程序,步骤会稍微复杂一些。

4.1 配置免密 SSH 登录

ssh-keygen -t rsa -b 2048
ssh-copy-id user@remote_host

4.2 运行多节点 MPI 程序

192.168.1.1 slots=2 192.168.1.2 slots=2
mpirun --hostfile hosts -np 4 ./hello_mpi

5. MPI 编译器封装器

MPI 提供了编译器封装器,使编译 MPI 程序更简单:

这些封装器自动处理 MPI 头文件和库的链接,可以直接用它们来编译 MPI 程序。

6. 常用 MPI 命令

7. 验证 MPI 环境

你可以使用 OpenMPI 自带的测试工具来确认 MPI 环境是否配置正确:

mpirun --host localhost -np 4 hostname

该命令将在本地机器上运行 4 个 MPI 进程,并输出机器名称。如果环境配置正确,应该会输出本地主机名 4 次。

总结

以上就是在Ubuntu上安装和配置MPI环境的详细步骤的详细内容,更多关于Ubuntu安装和配置MPI的资料请关注脚本之家其它相关文章!

您可能感兴趣的文章:
阅读全文