可耐石头

文章
5
资源
0
加入时间
3年0月20天

pytorch多进程部署遇到问题记录

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