今天使用multiprocessing.Pool
,發(fā)現(xiàn)程序在兩個(gè)大矩陣計(jì)算np.dot
時(shí)卡住不運(yùn)行会通,查了下發(fā)現(xiàn)以下設(shè)置后可正常運(yùn)行
import multiprocessing as mp
p.set_start_method("spawn")
with mp.Pool(args.num_jobs) as p:
p.map(YOUR_FUNCTION, PARAMS)
參考資料:
python multiprocessing子進(jìn)程hangs問(wèn)題 - 知乎 (zhihu.com)