阅读背景:

[tensorflow] tensorflow.***.ResourceExhaustedError: OOM when allocating tensor with shape[***]_haiqiang0225的博客

来源:互联网 

因为现在做的一个项目需要同时加载多个模型,而给的显卡又太渣了,加载不了几个就OOM了,查资料发现可以使用cpu来预测,使用cpu时是使用内存而不是使用显存,正好给的服务器的内存很大,所以这里记录一下因为现在做的一个项目需要同时加载多个模型,而给的显卡又太渣了,加载不了几个就OO




你的当前访问异常,请进行认证后继续阅读剩余内容。

分享到: