Tensorflow内存不足

时间:2019-11-16 05:31:18

标签: tensorflow

我正在尝试使用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 

1 个答案:

答案 0 :(得分:0)

答案是获得16GB内存GPU。