美文网首页
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