我是靠谱客的博主 玩命爆米花，最近开发中收集的这篇文章主要介绍[tensorflow] tensorflow.***.ResourceExhaustedError: OOM when allocating tensor with shape[***]，觉得挺不错的，现在分享给大家，希望可以做个参考。
概述
因为现在做的一个项目需要同时加载多个模型，而给的显卡又太渣了，加载不了几个就OOM了，查资料发现可以使用cpu来预测，使用cpu时是使用内存而不是使用显存，正好给的服务器的内存很大，所以这里记录一下
- tensorflow版本：2.4.2
- 显卡：NVIDIA Quadro P620
- 显存：2G
- 内存：128G
- 解决：执行
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"`
之后预测会使用cpu计算而不是gpu，会使用内存而不是使用显存。
最后
以上就是玩命爆米花为你收集整理的[tensorflow] tensorflow.***.ResourceExhaustedError: OOM when allocating tensor with shape[***]的全部内容，希望文章能够帮你解决[tensorflow] tensorflow.***.ResourceExhaustedError: OOM when allocating tensor with shape[***]所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
![bug6-ResourceExhaustedError: OOM when allocating tensor with shape[512,15,15,128]](/uploads/reation/bcimg14.png)
![OOM when allocating tensor with shape[5184,4,4,2048] and type float on /job:localhost/replica:0/task](/uploads/reation/bcimg1.png)
![tensorflow:OOM when allocating tensor with shape[225,256,256,36] and type float on /job:localhost/re](/uploads/reation/bcimg2.png)
![[tensorflow] tensorflow.***.ResourceExhaustedError: OOM when allocating tensor with shape[***]](/uploads/reation/bcimg4.png)
![Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos](/uploads/reation/bcimg7.png)
发表评论 取消回复