运行Tensorflow时GPU未分配内存

时间:2019-07-15 12:51:38

标签: python tensorflow deep-learning gpu

尝试使用Luminoth在Tensorflow中训练Fastrcnn和SSD对象检测模型。但是通常当我使用Tensorflow时,Tensorflow会完全使用GPU内存,但是现在它训练的任何模型都仅使用147MB。

我正在使用Ubuntu 16.04和python 3.6.8

enter image description here

有人可以告诉我可能的原因和解决方案吗?如果我错过了任何内容,或者过分强调或强调了一点,请在评论中告诉我。

0 个答案:

没有答案