今天使用multiprocessing.Pool,發(fā)現(xiàn)程序在兩個大矩陣計算np.dot時卡住不運行,查了下發(fā)現(xiàn)以下設置后可正常運行
import multiprocessing as mp
p.set_start_method("spawn")
with mp.Pool(args.num_jobs) as p:
p.map(YOUR_FUNCTION, PARAMS)
今天使用multiprocessing.Pool,發(fā)現(xiàn)程序在兩個大矩陣計算np.dot時卡住不運行,查了下發(fā)現(xiàn)以下設置后可正常運行
import multiprocessing as mp
p.set_start_method("spawn")
with mp.Pool(args.num_jobs) as p:
p.map(YOUR_FUNCTION, PARAMS)