原文:OOM when allocating tensor of shape [] and type float [[node conv2d_224/kernel/Initializer/random_uniform/min (defined at ./intances/utils.py:19) ]]

当你们在用模型加载,tensorflow gpu预测的时候,有没有出现这样的错误 经过网上的资料查阅,更多的解释是GPU的显存不足造成的,于是乎,我查看了一下GPU显存:nvidia smi 不看不知道,一看确实是那么回事:tensorflow gpu执行默认将GPU显存占满,官网也做了解释说明:https: www.tensorflow.org guide using gpu 刚开始没有看到官 ...

2019-05-29 11:34 0 757 推荐指数:

查看详情

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM