我是靠谱客的博主 玩命爆米花，这篇文章主要介绍[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]内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
![bug6-ResourceExhaustedError: OOM when allocating tensor with shape[512,15,15,128]](https://file2.kaopuke.com:8081/files_image/reation/bcimg14.png)
![OOM when allocating tensor with shape[5184,4,4,2048] and type float on /job:localhost/replica:0/task](https://file2.kaopuke.com:8081/files_image/reation/bcimg1.png)
![tensorflow:OOM when allocating tensor with shape[225,256,256,36] and type float on /job:localhost/re](https://file2.kaopuke.com:8081/files_image/reation/bcimg2.png)
![[tensorflow] tensorflow.***.ResourceExhaustedError: OOM when allocating tensor with shape[***]](https://file2.kaopuke.com:8081/files_image/reation/bcimg4.png)
![Resource exhausted: OOM when allocating tensor with shape[620,20000] and type float on /job:localhos](https://file2.kaopuke.com:8081/files_image/reation/bcimg7.png)
发表评论 取消回复