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跑------快
tensorflow-gpu验证准确率是报错如上:
解决办法:
1. 加入os.environ['CUDA_VISIBLE_DEVICES']='2'
强制使用CPU验证-----慢
2.'batch_size', 降低为32,即可使用GPU跑------快