13518219792
这篇文章将为大家详细讲解有关pytorch多进程如何优化，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。
目标：优化代码，利用多进程，进行近实时预处理、网络预测及后处理：
尝试了pytorch的multiprocessing，进行多进程同步处理以上任务。
from torch.multiprocessing import Pool,Manager
为了进行各进程间的通信，使用Queue，作为数据传输载体。
manager = Manager() input_queue = manager.Queue() output_queue = manager.Queue() show_queue = manager.Queue()
即将预处理进程处理后的图像放进 input_queue，而网络预测进程实时获取 input_queue队列中的数据，一旦放入，就从中取出，输入网络：
while 1： input = input_queue.get()
同理，将网络输出放入output_queue，再由后处理进程实时获取并进行后处理，处理后，放入show_queue
从而实现了多进程同步进行预处理、网络预测及后处理，加速了网络实时预测的表现。
问题及解决方案：
1. pytorch cuda报错，re-initialization报错问题：
习惯了在一开始将模型先加载进来放入gpu中，所以模型在主线程就完成了初始化，但是调用网络是在网络预测子进程进行的，就会导致跨进程重复初始化失败。
解决方案：
直接在子进程开始时初始化，其他进程可以设置个延时，等网络初始化好后再开始运行。
同理，数据输入网络也是同样的在子进程进行。
2. python3多进程编程，子进程不报错问题：
一开始总是代码运行起来什么都不出现，就开始各种debug，但因为python3中子进程不报错，出错了就卡在那里，就每次自己找bug很麻烦，所以就急需子进程的报错信息。
解决方案：
用try except组合，来打印子进程中某段程序错误，如下：
try: out = forward(input_img,model,1) except Exception as error: print(error)
3.进程完全不运行时，考虑是否是输入设置的问题，即当单变量输入时，后面要加都好，如：
pool.apply_async(load_frame,args=(input_queue,))
如果是pool.apply_async(load_frame,args=(input_queue))则该进程不会启动运行。
4. 一开始想优化cv2.resize，想用gpu下的tensor的resize_代替，但发现这种方式和numpy.resize一脉相承啊，根本不是我们想要的resize，如果是变大的话，这种resize会直接按顺序填，然后剩下的就填0，实在是太草率了。。。
解决方案：还没有很好的替代方案，只找了一下，cuda::resize，但是好像貌似没有python接口，要是混合编程好像有点小题大做，得不偿失了。
关于“pytorch多进程如何优化”这篇文章就分享到这里了，希望以上内容可以对大家有一定的帮助，使各位可以学到更多知识，如果觉得文章不错，请把它分享出去让更多的人看到。
