paralelización multiplicación de cálculo vectores-como en pitón

Question

I tienen un trozo de código como

for i in range(0, len(a)) 
    b[i] = func(a[i]) 

donde a y b son matrices de la misma longitud, se le da una (y grande), func es una función eso tiene muchas variables locales pero no usa ninguna variable global.

Me gustaría distribuir cálculos de func en varias CPU. Presumiblemente necesito usar el módulo de multiprocesamiento, pero no he encontrado ningún ejemplo relevante. ¿Podrías ayudar? Gracias.

Answer 1

Véase el primer ejemplo de código en el multiprocessing docs:

from multiprocessing import Pool 

# you could define `func`, `a` here 

if __name__=="__main__": 
    p = Pool() # use all available CPU cores 
    b = p.map(func, a) 

Answer 2

Usar grupo de proceso. Puede ver una muestra completa en mi github: https://github.com/mariazverina/codejam/blob/master/src/main.py

from multiprocessing import Pool 

p = Pool(4) # set to number of cores 
b = p.map(func, a)