【并行计算】Hello, world! —— 我的第一个并行程序

【并行计算】Hello, world! —— 我的第一个并行程序

在2019年大数据与人工智能的暑期学校中,我第一次接触到了并行编程,当时使用的是长沙超算中心的天河一号,大概同时使用过128个核心,一起来做些简单的运算,后来也在自己的笔记本上做过4-8个线程的简单尝试,最近又重新捡了起来,温习一下,并做以记录。

1.程序代码:

  话不多所,先上代码,毕竟Talking is cheap, show me the code!

from mpi4py import MPI
comm = MPI.COMM_WORLD
rank = comm.Get_rank()
print('Hello, world! This is from process: %d' % rank)

我们将文件保存为MPI_Helloworld.py

2.代码介绍

1.我们首先导入mpi4py库中的MPI模块

from mpi4py import MPI

2.将comm(communication的简写)定义为一组可以互相发送消息的进程集合

comm = MPI.COMM_WORLD

3.获得正在调用进程的进程号

rank = comm.Get_rank()

4.各进程各自打印输出

print('Hello, world! This is from process: %d' % rank)

3.执行

  在终端执行:

mpiexec -n 4 python MPI_Helloworld.py

  可以看到输出

fire@fire-P7xxTM1:~/multi_process_thread$ mpiexec -n 4 python mpi_helloworld.py 
Hello, world! This is from process: 0
Hello, world! This is from process: 1
Hello, world! This is from process: 2
Hello, world! This is from process: 3
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章