在使用 Apache JMeter 进行性能测试期间,我面临着初始模型的问题。
错误:分配形状为 [800,1280,3] 和类型的张量时出现 OOM
通过分配器浮动在 /job:localhost/replica:0/task:0/device:GPU:0 上
GPU_0_bfc [[节点:Cast = CastDstT=DT_FLOAT,SrcT=DT_UINT8,
_device="/job:localhost/replica:0/task:0/device:GPU:0"]]
提示:如果您想在 OOM 发生时查看分配的张量列表,
将 report_tensor_allocations_upon_oom 添加到当前的 RunOptions
分配信息。
OOM 代表内存不足。这意味着您的 GPU 空间不足,可能是因为您分配了太大的其他张量。您可以通过缩小模型或减小批量大小来解决此问题。从外观上看,您正在输入大图像(800x1280),您可能需要考虑下采样。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)