欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Tensorflow/Keras Linux上多进程加载模型卡死解决办法

程序员文章站 2022-03-03 14:07:06
...

因为在Windows上是没有问题的,因此考虑可能是进程复制的方式有问题!
多进程文档

查看文档后,强制使用spawn方式,问题解决!

在代码最开始的地方,加入如下语句

import multiprocessing
if __name__ == '__main__':
    multiprocessing.set_start_method('spawn')