GPU在运行tens时未分配内存

2024-06-08 22:48:03 发布

您现在位置:Python中文网/ 问答频道 /正文

尝试用Luminoth在Tensorflow中训练Fastrcnn和SSD目标检测模型。但通常当我使用Tensorflow时,GPU内存将完全由Tensorflow使用,但现在只有147MB被用于它正在训练的任何模型

我使用的是ubuntu16.04和python 3.6.8

enter image description here

有人能告诉我可能的原因和解决办法吗?如果我漏掉了什么,过分强调或不强调某一点,请在评论中告诉我


Tags: 内存模型目标gputensorflow评论原因ssd