【问题标题】:Tensorflow ran out of memoryTensorFlow 内存不足
【发布时间】:2020-03-12 05:57:47
【问题描述】:

我正在尝试使用 ssd_mobilenet_v2_quantized_300x300_coco.config 进行训练

当我尝试训练时它会显示这个

W tensorflow/core/common_runtime/bfc_allocator.cc:239] 分配器 (GPU_0_bfc) 用 freed_by_count=0 分配 927.72MiB 时内存不足。调用者表示这不是失败,但可能意味着如果有更多可用内存可能会提高性能。

这些是我用来训练的命令

python train.py --pipeline_config_path=training/ssd_mobilenet_v2_quantized_300x300_coco.config --train_dir=training/ --alsologtostderr 

【问题讨论】:

  • 仅供参考,如果我使用 ssd_mobilenet_v2_coco.config 这个问题就不会发生
  • 什么GPU?配置说明了什么?
  • @geometrikal GTX 1080 ti

标签: tensorflow


【解决方案1】:

答案是获得 16GB 显存的 GPU。

【讨论】:

    猜你喜欢
    • 2020-08-06
    • 1970-01-01
    • 2019-06-06
    • 2018-11-17
    • 1970-01-01
    • 1970-01-01
    • 2017-01-13
    • 2018-12-13
    • 2014-10-15
    相关资源
    最近更新 更多