python - 在多个内核上运行 python 程序

Question

我是多核编程的新手。下面的程序只使用了一个内核。我怎样才能让它在多个内核上运行(我有 4 个内核)。

simDict={}

def sim(outer,inner, ...):
    val= /*do some math*/
    simDict[...]=val

def foo():
   for outer in xrange(0, limit):
      for inner in xrange(outer, limit):
          sim(outer,inner, ...)
foo()

Answer 1

简单:

from multiprocessing import Pool

p = Pool()

def do_inner(outer, limit):
    for inner in xrange(outer, limit):
        sim(outer, inner, ...)

def foo():
    p.map(do_inner, xrange(limit))

foo()

这使用 multiprocessing.Pool 创建工作进程池。