python-3.x - 错误 : OOM when allocating tensor with shape

Question

在使用 Apache JMeter 进行性能测试期间，我的初始模型遇到了问题。

Error: OOM when allocating tensor with shape[800,1280,3] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [[Node: Cast = CastDstT=DT_FLOAT, SrcT=DT_UINT8, _device="/job:localhost/replica:0/task:0/device:GPU:0"]] Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.

Answer 1

OOM 代表内存不足。这意味着您的 GPU 空间不足，大概是因为您分配了其他太大的张量。您可以通过缩小模型或减少批量大小来解决此问题。从它的外观来看，您正在输入一张大图像 (800x1280)，您可能需要考虑下采样。