概述
pytorch 多进程初始化问题
pytorch 进行多进程部署时，遇到cuda initial error 。解决办法：把模型初始化不要放在__init__函数中，并且在主函数中加上mp.set_start_method(‘spawn’)。
注释：这是我在实际测pytorch模型时，遇到的问题，是这样解决的，其他的情况可能会不同。
最后
以上就是可耐石头为你收集整理的pytorch多进程部署遇到问题记录的全部内容，希望文章能够帮你解决pytorch多进程部署遇到问题记录所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复