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

显存不够----ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[4096]

时间:2018-02-08 12:19:16      阅读:1894      评论:0      收藏:0      [点我收藏+]

标签:resource   csharp   ati   erro   sdn   art   mod   charles   sharp   

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[4096]

 

类似问题 https://github.com/CharlesShang/TFFRCNN/issues/68

参考:

 http://blog.csdn.net/qing101hua/article/details/77153222

方案:

 

take a look at the memory usage of your gpu with

nvidia-smi -l 3

  

if still free memory that can be used
-> edit lib/fast_rcnn/train.py
~line396 : config.gpu_options.per_process_gpu_memory_fraction = 0.40
edit 0.40 to larger propotion (e.g. 0.9)

显存不够----ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[4096]

标签:resource   csharp   ati   erro   sdn   art   mod   charles   sharp   

原文地址:https://www.cnblogs.com/leoking01/p/8430604.html

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