繁体   English   中英

在python中的并行进程中运行类方法

[英]Run a class method in parallel processes in python

我正在尝试并且未能并行运行一个巨大的循环。 循环恰好是特定类的一种方法,在循环内部我称之为另一种方法。 它确实可以工作,但是由于某种原因,列表中只有一个进程,并且输出(请参见代码)始终为“ Worker 0”。 要么未创建进程,要么进程未并行运行。 结构如下:

main.py

from my_class.py import MyClass

def main():
    class_object = MyClass()
    class_object.method()

if __name__ == '__main__':
    main()

my_class.py

from multiprocessing import Process

MyClass(object):
    def __init__(self):
        # do something

    def _method(self, worker_num, n_workers, amount, job, data):
        for i, val in enumerate(job):
            print('Worker %d' % worker_num)
            self.another_method(val, data)

    def another_method(self):
        # do something to the data

    def method(self):
        # definitions of data and job_size go here

        n_workers = 16
        chunk = job_size // n_workers
        resid = job_size - chunk * n_workers

        workers = []
        for worker_num in range(n_workers):
            st = worker_num * chunk
            amount = chunk if worker_num != n_workers - 1 else chunk + resid
            worker = Process(target=self._method, args=[worker_num, n_workers, amount, job[st:st+amount], data])
            worker.start()
            workers.append(worker)

        for worker in workers:
            worker.join()

        return data

我已经阅读了一些有关需要主模块可导入的子进程的信息,但是我不知道该如何做。

问题 :...但是仍然只使用一个内核。 所以问题是,我可以将多个核与Process对象一起使用

这并不取决于Python解释器哪个Process使用哪个CPU。
相关: 在什么CPU核心是我的Python进程运行

用以下内容扩展def _method(... ,以查看实际发生的情况:

注意getpidcore(pid)分发的依赖项,可能会失败

def getpidcore(pid):
    with open('/proc/{}/stat'.format(pid), 'rb') as fh:
        core = int(fh.read().split()[-14])
        return core

class MyClass(object): 
    ...
    def _method(self, worker_num, n_workers, amount, job, data):
        for i, val in enumerate(job):
            core = getpidcore(os.getpid())
            print('core:{} pid:{} Worker({})'.format(core, os.getpid(), (worker_num, n_workers, amount, job)))

输出

 core:1 pid:7623 Worker((0, 16, 1, [1])) core:1 pid:7625 Worker((2, 16, 1, [3])) core:0 pid:7624 Worker((1, 16, 1, [2])) core:1 pid:7626 Worker((3, 16, 1, [4])) core:1 pid:7628 Worker((5, 16, 1, [6])) core:0 pid:7627 Worker((4, 16, 1, [5])) 

经过Python测试:Linux上为3.4.2

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM