美文网首页
tensorflow错误:显存/内存不够

tensorflow错误:显存/内存不够

作者: 可可酱X3 | 来源:发表于2020-05-27 20:57 被阅读0次

原文链接:https://blog.csdn.net/Muzi_Water/java/article/details/88390840

原文链接:https://blog.csdn.net/wanglitao588/java/article/details/77033659


tensorflow错误:

“tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[32,256,55,55]”

解决方法:

减少批处理Batch的大小

降低全连接层的维度

增加池化层

缩小输入图片大小

总之呢,就是显卡内存不能满足此次训练,所以就是要降低所需内存要求。


错误:InternalError: Dst tensor is not initialized.

分析:

出现这个错误一般是GPU内存耗尽,挂在后台进程中的IPython内核会这样做

解决办法:

运行程序之前,先运行export CUDA_VISIBLE_DEVICES=1,仅显卡设备1GPU可见,tensorflow 算是一个比较贪心的工具了就算用device_id指定gpu 也会占用别的GPU的显存资源 必须在执行程序前执行

export CUDA_VISIBLE_DEVICES=n(n为可见的服务器编号)再去执行Python 代码.py 才不会占用别的GPU资源

相关文章

网友评论

      本文标题:tensorflow错误:显存/内存不够

      本文链接:https://www.haomeiwen.com/subject/epigahtx.html