ufscar/ubuntu_ompi Docker 镜像 - 轩辕镜像

Ubuntu with OpenMPI 镜像文档

镜像概述

本镜像基于Ubuntu操作系统，预集成OpenMPI并行计算框架，提供完整的并行计算运行环境。OpenMPI是一个开源的消息传递接口（MPI）实现，支持跨节点通信与分布式计算，适用于构建高性能计算（HPC）应用。

核心功能与特性

• 基础环境：基于最新稳定版Ubuntu，包含完整的Linux系统工具链
• 预配置OpenMPI：已安装OpenMPI库及开发文件，支持MPI 1/2/3标准
• 编译支持：集成GCC编译器，可直接编译C/C++ MPI程序
• 多节点通信：支持容器间网络通信，可构建分布式MPI集群
• 轻量级设计：在保证功能完整的前提下优化镜像体积

使用场景

• 科学计算应用开发与调试
• 分布式并行算法验证
• 高性能计算任务原型测试
• 教学环境中的并行计算实践
• 多节点MPI程序性能评估

使用方法

基本运行方式

通过以下命令启动交互式容器：

bash
docker run -it --name mpi-instance ubuntu-openmpi /bin/bash

编译与运行MPI程序

1. 在容器内创建MPI测试程序（以C语言为例）：

   bash
   cat > hello_mpi.c << EOF
   #include <mpi.h>
   #include <stdio.h>
   
   int main(int argc, char** argv) {
       MPI_Init(&argc, &argv);
       int rank, size;
       MPI_Comm_rank(MPI_COMM_WORLD, &rank);
       MPI_Comm_size(MPI_COMM_WORLD, &size);
       printf("Hello from rank %d of %d\n", rank, size);
       MPI_Finalize();
       return 0;
   }
   EOF
   

2. 使用mpicc编译程序：

   bash
   mpicc hello_mpi.c -o hello_mpi
   

3. 本地运行（单节点多进程）：

   bash
   mpirun -np 4 ./hello_mpi  # 使用4个进程运行
   

多节点MPI集群配置（docker-compose示例）

创建docker-compose.yml配置文件：

yaml
version: '3'
services:
  mpi-node1:
    image: ubuntu-openmpi
    container_name: mpi-node1
    tty: true
    networks:
      mpi-net:
        ipv4_address: 172.20.0.2

  mpi-node2:
    image: ubuntu-openmpi
    container_name: mpi-node2
    tty: true
    networks:
      mpi-net:
        ipv4_address: 172.20.0.3

networks:
  mpi-net:
    driver: bridge
    ipam:
      config:
        - subnet: 172.20.0.0/24

启动集群：

bash
docker-compose up -d

在主节点（如mpi-node1）运行跨节点MPI程序：

bash
# 在mpi-node1容器内执行
mpirun -np 4 -hosts 172.20.0.2,172.20.0.3 ./hello_mpi

环境配置说明

• OpenMPI安装路径：/usr/lib/openmpi
• 头文件路径：/usr/include/openmpi
• 默认MPI通信端口：***（可通过--mca btl_tcp_port_min和--mca btl_tcp_port_max调整）
• 支持的MPI传输协议：TCP/IP、共享内存（默认启用）

注意事项

• 多节点通信需确保容器间网络互通，推荐使用自定义桥接网络
• 运行多节点任务时需配置SSH免密登录或使用MPI的rsh/ssh启动方式
• 资源密集型任务建议通过--cpus和--memory参数限制容器资源