OOM when allocating tensor of shape [] and type float [[node conv2d_224/kernel/Initializer/random_un...
当你们在用模型加载,tensorflow-gpu预测的时候,有没有出现这样的错误??经过网上的资料查阅,更多的解释是GPU的显存不足造成的,于是乎,我查看了一下GPU显存:nvidia-smi不看不知道,一看确实是那么回事:tensorflow-gpu执行默认将GPU显存占满,官网也做了解释说明:https://www.tensorflow.org/guide/usin...