pytorch 多进程初始化问题
pytorch 进行多进程部署时，遇到cuda initial error 。解决办法：把模型初始化不要放在__init__函数中，并且在主函数中加上mp.set_start_method(‘spawn’)。
注释：这是我在实际测pytorch模型时，遇到的问题，是这样解决的，其他的情况可能会不同。
最后
以上就是可耐石头最近收集整理的关于pytorch多进程部署遇到问题记录的全部内容，更多相关pytorch多进程部署遇到问题记录内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复