OOM when allocating tensor of shape [] and type float [[node conv2d_224/kernel/Initializer/random_un...

66 阅读 0 评论 44 点赞

我是靠谱客的博主忧伤刺猬，最近开发中收集的这篇文章主要介绍OOM when allocating tensor of shape [] and type float [[node conv2d_224/kernel/Initializer/random_un...，觉得挺不错的，现在分享给大家，希望可以做个参考。

当你们在用模型加载，tensorflow-gpu预测的时候，有没有出现这样的错误？？

经过网上的资料查阅，更多的解释是GPU的显存不足造成的，于是乎，我查看了一下GPU显存：nvidia-smi

不看不知道，一看确实是那么回事：tensorflow-gpu执行默认将GPU显存占满，官网也做了解释说明：https://www.tensorflow.org/guide/using_gpu

刚开始没有看到官网的解释，思路是当程序执行完成之后，能不能自动的执行释放显存的程序，网上这方面的代码不多，后来又看到上面官网的解释，显存不能释放，可能会有更严重的内存碎片，因此程序释放显存的想法就打消了。

在仔细看官网的解释，可能比较好的方法是指定GPU所占的比例，于是乎，我又多次的尝试：一般来说，这些指定的过程都放在代码的开始处&

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

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