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

错误TF2。完成GeneratorDataset迭代器时出错。Python解释器状态未初始化—它不会发生在TF1中,python,tensorflow,keras,Python,Tensorflow,Keras,这个错误非常奇怪,让我抓狂,因为它在Tensorflow 1.15中没有提示。我创建了一个数据生成器,在使用Tensorflow 1.15时可以完美地工作,但因为我有一些 在TF1中无法处理的内存不足问题,请升级到TF2,我在培训开始时收到此消息 我使用的HPC具有特斯拉V100-PCIE-16GB。似乎GPU无法加载python解释器Python3.7 有什么想法吗?正如我所说,TF1工作得非常完美 2021-05-20 17:15:31.744926: W tensorflow/core

这个错误非常奇怪,让我抓狂,因为它在Tensorflow 1.15中没有提示。我创建了一个数据生成器,在使用Tensorflow 1.15时可以完美地工作,但因为我有一些 在TF1中无法处理的内存不足问题,请升级到TF2,我在培训开始时收到此消息

我使用的HPC具有特斯拉V100-PCIE-16GB。似乎GPU无法加载python解释器Python3.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}}]]




这回答了你的问题吗?添加此代码段
gpus=tf.config.list\u physical\u devices('GPU')
tf.config.experimental.set\u memory\u growth(GPU[0],True)
。谢谢