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'
文章出处登录后可见！
已经登录？立即刷新