在异步执行之前启动多处理池映射/星图?

时间:2019-12-25 13:50:14

标签: python multiprocessing environment-variables pool

我可以使用4个进程执行launch_training_job函数len(experiments)

p = Pool(4)
p.starmap(launch_training_job, experiments)

但是我想根据正在执行的进程来相应执行

os.environ['CUDA_VISIBLE_DEVICES'] = 0
os.environ['CUDA_VISIBLE_DEVICES'] = 1
os.environ['CUDA_VISIBLE_DEVICES'] = 2
os.environ['CUDA_VISIBLE_DEVICES'] = 3

在进程开始执行功能之前。在调用该函数之前,我真的找不到如何添加初始化例程的示例或文档。

我该怎么做?

0 个答案:

没有答案