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