码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2019-07-22 13:52:29      阅读:725      评论:0      收藏:0      [点我收藏+]

标签:img   fail   vpd   http   text   sdn   ext   uda   log   

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

解决办法:

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

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

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

 

技术图片

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

标签:img   fail   vpd   http   text   sdn   ext   uda   log   

原文地址:https://www.cnblogs.com/jfdwd/p/11225196.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!