Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos

106 阅读 0 评论 70 点赞

我是靠谱客的博主丰富小猫咪，最近开发中收集的这篇文章主要介绍Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

在CPU下跑的时候并没有报错，换成GPU后就一直OOM，后来被提醒道，显存比内存小很多啊，所以改小了batch_size,并将网络结构进行了优化，减少了词向量的维度和隐藏层节点个数。

另外，在代码里添加了GPU的相关代码：

 config = tf.ConfigProto(allow_soft_placement=True)
 config.gpu_options.allocator_type = 'BFC'
 config.gpu_options.per_process_gpu_memory_fraction = 0.80

新增的部分红线圈出：

修改完之后就不报OOM的错误了。

其他可参考：https://github.com/CharlesShang/TFFRCNN/issues/68

最后

以上就是丰富小猫咪为你收集整理的Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos的全部内容，希望文章能够帮你解决Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。

本文分类：Other
浏览次数：106 次浏览
发布日期：2023-10-23 01:31:36
本文链接：https://www.kaopuke.com/article/k-p-k_13_u_23_o_22_fz_13__7__14_4.html

Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos

概述

最后

评论列表共有 0 条评论

发表评论取消回复

Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos

概述

最后

相关文章

评论列表共有 0 条评论

发表评论 取消回复

发表评论取消回复