OP_REQUIRES failed at conv_ops.cc:386 : Resource exhausted: OOM when allocating tensor with shape..

tensorflow-gpu验证准确率是报错如上：

解决办法：

1. 加入os.environ['CUDA_VISIBLE_DEVICES']='2'

强制使用CPU验证-----慢

2.'batch_size', 降低为32，即可使用GPU跑------快

 

posted @ 2019-07-22 12:43  交流_QQ_2240410488  阅读(4971)  评论(0)  编辑  收藏  举报