错误: OOM分配带有形状的张量
错误: OOM分配带有形状的张量
提问于 2018-06-08 20:20:44
在使用Apache JMeter进行性能测试时，我遇到了初始模型的问题。
错误: OOM当分配具有形状800,1280，3的张量，并通过分配器GPU__bfc在/job:localhost/replica:0/task:0/device:GPU:0键入float时[Node: Cast = CastDstT=DT_FLOAT，SrcT=DT_UINT8，Node提示:如果您希望在发生OOM时查看已分配张量的列表，请将report_tensor_allocations_upon_oom添加到RunOptions以获取当前分配信息。
回答 2
Stack Overflow用户
发布于 2018-06-09 00:31:49
OOM代表内存不足。这意味着你的GPU已经用完了空间，大概是因为你分配了其他太大的张量。您可以通过使模型更小或减小批处理大小来解决此问题。从外观上看，您正在输入一个大图像(800x1280)，您可能需要考虑下采样。
Stack Overflow用户
发布于 2018-06-09 03:28:25
如果您手头有多个GPU，请选择一个不像此GPU那样繁忙的GPU(可能的原因，其他进程也在此GPU上运行)。转到终端并键入
export CUDA_VISIBLE_DEVICES=1 其中1是其他可用GPU的数量。重新运行相同的代码。
您可以使用以下命令检查可用的GPU
nvidia-smi 这将显示哪些GPU可用，以及每个GPU上有多少可用内存
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/50760543
复制相关文章
相似问题
腾讯云开发者
