是什么导致了 python 中 map.pool 的使用被挂起？

Question

我正在运行一个命令行程序，并通过管道输入文本作为参数：

somecommand.exe < someparameters_tin.txt

它运行一段时间（通常是一小时到几个小时），然后将结果写入多个文本文件中。我正在尝试编写一个脚本来同时启动其中几个，使用多核机器上的所有核心。在其他操作系统上，我会分叉，但这在 Windows 的许多脚本语言中都没有实现。 Python 的多处理看起来可能会成功，所以我想尝试一下，尽管我根本不了解 python。我希望有人能告诉我我做错了什么。

我编写了一个脚本（如下），我指向一个目录，如果找到可执行文件和输入文件，并使用 pool.map 和 n 池以及使用 call 的函数启动它们。我看到的是，最初（启动第一组 n 个进程）似乎很好，100% 使用 n 个核心。但随后我发现进程处于空闲状态，没有使用或仅使用了百分之几的 CPU。那里总是有 n 个进程，但它们没有做太多事情。当他们去写入输出数据文件时，似乎会发生这种情况，一旦启动，一切都会陷入困境，总体核心利用率范围从几个百分点到偶尔达到 50-60% 的峰值，但永远不会接近 100%。

如果我可以附加它（编辑：我不能，至少现在），这是进程的运行时间图。较低的曲线是当我打开 n 个命令提示符并手动保持 n 个进程同时运行时，轻松地将计算机保持在 100% 附近。 （该线是有规律的，在 32 个不同的进程中改变参数，从接近 0 小时缓慢增加到 0.7 小时。）上面的线是该脚本的某些版本的结果 - 运行时间平均增加了约 0.2 小时，并且是更难以预测，就像我在底线中添加了 0.2 + 一个随机数。

这是情节的链接： 运行时图

编辑：现在我想我可以添加该图。

我做错了什么？

from multiprocessing import Pool, cpu_count, Lock
from subprocess import call
import glob, time, os, shlex, sys
import random

def launchCmd(s):
    mypid = os.getpid()
    try:
        retcode = call(s, shell=True)
        if retcode < 0:
            print >>sys.stderr, "Child was terminated by signal", -retcode
        else:
            print >>sys.stderr, "Child returned", retcode
    except OSError, e:
        print >>sys.stderr, "Execution failed:", e

if __name__ == '__main__':

    # ******************************************************************
    # change this to the path you have the executable and input files in
    mypath = 'E:\\foo\\test\\'
    # ******************************************************************

    startpath = os.getcwd()
    os.chdir(mypath)
    # find list of input files
    flist = glob.glob('*_tin.txt')
    elist = glob.glob('*.exe')
    # this will not act as expected if there's more than one .exe file in that directory!
    ex = elist[0] + ' < '

    print
    print 'START'
    print 'Path: ', mypath
    print 'Using the executable: ', ex
    nin = len(flist)
    print 'Found ',nin,' input files.'
    print '-----'
    clist = [ex + s for s in flist]
    cores = cpu_count()
    print 'CPU count ', cores
    print '-----'

    # ******************************************************
    # change this to the number of processes you want to run
    nproc = cores -1
    # ******************************************************

    pool = Pool(processes=nproc, maxtasksperchild=1)    # start nproc worker processes
    # mychunk = int(nin/nproc)      # this didn't help
    # list.reverse(clist)           # neither did this, or randomizing the list
    pool.map(launchCmd, clist)      # launch processes
    os.chdir(startpath)             # return to original working directory
    print 'Done'

Answer 1

这些进程是否有可能尝试写入公共文件？在 Linux 下，它可能会正常工作，破坏数据，但不会减慢速度；但在 Windows 下，一个进程可能会获取该文件，而所有其他进程可能会挂起，等待该文件变得可用。

如果您用一些使用 CPU 但不写入磁盘的愚蠢任务替换实际任务列表，问题是否会重现？例如，您可能有计算某个大文件的 md5sum 的任务；一旦文件被缓存，其他任务将是纯 CPU 的，然后是单行输出到 stdout。或者计算一些昂贵的函数之类的。

Answer 2

我认为我知道这一点。当您调用map时，它会将任务列表分成每个进程的“块”。默认情况下，它使用足够大的块，以便可以向每个进程发送一个。这是基于所有任务需要大约相同的时间来完成的假设。

根据您的情况，完成这些任务可能需要不同的时间。因此，一些工作人员先于其他工作人员完成工作，而这些 CPU 则处于闲置状态。如果是这样的话，那么这应该按预期工作：

pool.map(launchCmd, clist, chunksize=1)

效率较低，但这应该意味着每个工作人员在完成时会获得更多任务，直到它们全部完成。