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..

相关文章:

  • 2021-10-17
  • 2022-01-14
  • 2022-12-23
  • 2021-07-17
  • 2022-12-23
  • 2021-07-18
  • 2021-07-08
  • 2022-12-23
猜你喜欢
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-09-19
  • 2021-11-16
  • 2022-12-23
  • 2021-08-07
相关资源
相似解决方案