在使用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以获取当前分配信息。
发布于 2018-06-09 00:31:49
OOM代表内存不足。这意味着你的GPU已经用完了空间,大概是因为你分配了其他太大的张量。您可以通过使模型更小或减小批处理大小来解决此问题。从外观上看,您正在输入一个大图像(800x1280),您可能需要考虑下采样。
发布于 2018-06-09 03:28:25
如果您手头有多个GPU,请选择一个不像此GPU那样繁忙的GPU(可能的原因,其他进程也在此GPU上运行)。转到终端并键入
export CUDA_VISIBLE_DEVICES=1
其中1是其他可用GPU的数量。重新运行相同的代码。
您可以使用以下命令检查可用的GPU
nvidia-smi
这将显示哪些GPU可用,以及每个GPU上有多少可用内存
https://stackoverflow.com/questions/50760543
复制相似问题