Debug: Error occurred when finalizing GeneratorDataset iterator: FAILED_PRECONDITION: Python interpr

tensorflow中出现的问题。话说真的是tf不友好的地方,很多地方的错误都能导致这个错误,从数据导入到模型结构,到显存。

可能原因多方面:

  • 网络模型定义有问题
  • 显卡设置显存限制,
physical_devices = tf.config.list_physical_devices('GPU') 
for device in physical_devices:
    tf.config.experimental.set_memory_growth(device, True)
  • 显存不足,减少batch_size
  • 检查输入数据形状与定义是否相符

清理内存,线上慎用

sudo sh -c 'sync && echo 3 > /proc/sys/vm/drop_caches'

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
社会演员多的头像社会演员多普通用户
上一篇 2022年5月17日
下一篇 2022年5月17日

相关推荐