pytorch 多进程初始化问题
pytorch 进行多进程部署时,遇到cuda initial error 。解决办法:把模型初始化不要放在__init__函数中,并且在主函数中加上mp.set_start_method(‘spawn’)。
注释:这是我在实际测pytorch模型时,遇到的问题,是这样解决的,其他的情况可能会不同。
pytorch 多进程初始化问题
pytorch 进行多进程部署时,遇到cuda initial error 。解决办法:把模型初始化不要放在__init__函数中,并且在主函数中加上mp.set_start_method(‘spawn’)。
注释:这是我在实际测pytorch模型时,遇到的问题,是这样解决的,其他的情况可能会不同。