今天小编就为大家分享一篇pytorch多进程加速及代码优化方法。具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

目标：优化代码，利用多进程，进行近实时预处理、网络预测及后处理： 本人尝试了pytorch的multiprocessing，进行多进程同步处理以上任务。 from torch.multiprocessing import Pool,Manager 为了进行各进程间的通信，使用Queue，作为数据传输载体。 manager = Manager() input_queue = manager.Queue() output_queue = manager.Queue() show_