python - 错误 TF2。完成 GeneratorDataset 迭代器时发生错误。Python解释器状态未初始化它不会发生在TF1中

Question

这个错误很奇怪，让我抓狂，因为它在 Tensorflow 1.15 中没有提示。我创建了一个在使用 Tensorflow 1.15 时可以完美运行的数据生成器，但是由于我在 TF1 中无法解决一些内存不足问题，因此升级到 TF2 并且我在培训开始时收到此消息。

我正在使用具有 Tesla V100-PCIE-16GB 的 HPC。似乎 GPU 无法加载 python 解释器 python 3.7。

有什么想法吗？正如我所说，TF1 完美运行。

2021-05-20 17:15:31.744926: W tensorflow/core/kernels/data/generator_dataset_op.cc:107] Error occurred when finalizing GeneratorDataset iterator: Failed precondition: Python interpreter state is not initialized. The process may be terminated.
     [[{{node PyFunc}}]]

标签： pythontensorflowkeras