Цикл выполняется для некоторого количества моделей (n_mod) и распределяется между n_cpu. Как вы заметили, запуск этого кода под именем mpirun -np 4 python test_mpi.py создает 4 индикатора выполнения. Это понятно. Но есть ли способ использовать tdqm, чтобы получить один индикатор выполнения, показывающий, сколько моделей было завершено?
from tqdm import tqdm
from mpi4py import MPI
import time
comm = MPI.COMM_WORLD
cpu_ind = comm.Get_rank()
n_cpu = comm.Get_size()
n_mod=100
for i in tqdm(range(n_mod)):
if (cpu_ind == int(i/int(n_mod/n_cpu))%n_cpu):
#some task here which is a function of i
time.sleep(0.02)
Подробнее здесь: https://stackoverflow.com/questions/791 ... job-python
Один индикатор выполнения для параллельного задания Python ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение