Tensorflow/Keras Linux上多进程加载模型卡死解决办法
程序员文章站
2022-03-03 14:07:06
...
因为在Windows上是没有问题的,因此考虑可能是进程复制的方式有问题!
多进程文档
查看文档后,强制使用spawn方式,问题解决!
在代码最开始的地方,加入如下语句
import multiprocessing
if __name__ == '__main__':
multiprocessing.set_start_method('spawn')
上一篇: 企业代码提交和发布流程
下一篇: oh-my-zsh 诡异字符问题