玩命爆米花

文章
7
资源
0
加入时间
4年1月23天
[tensorflow] tensorflow.***.ResourceExhaustedError: OOM when allocating tensor with shape[***]

[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，会使用内存而不是使用显存。.
遇到的坑 2023-10-23

他的专栏

电脑硬件（0）
Windows 7（0）
JavaScript（0）
漏洞分析（0）
Window服务器（1）
正则表达式（1）
PHP编程（1）
C#（1）
遇到的坑（1）
spring boot（1）
广告业务系统（1）

他的归档

2023年10月（1）

热门文章

1
WIN2008 R2 Active Directory 之二 部署企业中Windo
2
正则表达式口诀 正则表达式学习工具
3
php HtmlReplace输入过滤安全函数
4
C#编写控制台程序纸牌游戏
5
[tensorflow] tensorflow.***.ResourceExhaustedError: OOM when allocating tensor with shape[***]
6
class path resource [xx] cannot be resolved to absolute file path because it does not reside
7
批量将标签数据写入hbase中