我正在用Python做一个机器学习项目,所以我必须做并行预测功能,这是我在程序中使用的功能。
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
问题是我所有的CPU仅加载20-40%(总和为100%)。我使用multiprocessing.dummy,因为我在酸洗功能中的多处理模块有一些问题。
芜湖不芜
开心每一天1111
撒科打诨
相关分类