我可以使用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
在进程开始执行功能之前。在调用该函数之前,我真的找不到如何添加初始化例程的示例或文档。
我该怎么做?