未验证 提交 84c8096c 编写于 作者: J JingZhuangzhuang 提交者: GitHub

fix infer gpu strage (#41924)

上级 afaf58ac
......@@ -1077,6 +1077,12 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor<
process_level_allocator_enabled = true;
}
// TODO(Jingzhuangzhuang): Fix trt error when allocator_strategy is
// auto_growth
if (config.tensorrt_engine_enabled()) {
gflags.push_back("--allocator_strategy=naive_best_fit");
}
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.
先完成此消息的编辑!
想要评论请 注册