未验证 提交 39aeaa16 编写于 作者: W Wilber 提交者: GitHub

fix jetson problem (#30939)

上级 01ccfbcd
......@@ -653,6 +653,13 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor<
process_level_allocator_enabled = true;
}
// TODO(wilber): jetson tx2 may fail to run the model due to insufficient memory
// under the native_best_fit strategy. Modify the default allocation strategy to
// auto_growth. todo, find a more appropriate way to solve the problem.
#ifdef WITH_NV_JETSON
gflags.push_back("--allocator_strategy=auto_growth");
#endif
if (framework::InitGflags(gflags)) {
VLOG(3) << "The following gpu analysis configurations only take effect "
"for the first predictor: ";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册