multiprocessing.dummy en Python no utiliza 100% de CPU
Estoy haciendo un proyecto de aprendizaje automático en Python, así que tengo que hacer una función de predicción paralela, que estoy usando en mi programa.
from multiprocessing.dummy import Pool
from multiprocessing import cpu_count
def multi_predict(X, predict, *args, **kwargs):
pool = Pool(cpu_count())
results = pool.map(predict, X)
pool.close()
pool.join()
return results
El problema es que todas mis CPU se cargaron solo en 20-40% (en suma, es 100%). Utilizo multiprocessing.dummy porque tengo algunos problemas con el módulo de multiprocesamiento en la función de decapado.