未验证 提交 b39c0116 编写于 作者: Z Zhaolong Xing 提交者: GitHub

specify the auto growth allocator for inference. (#21448)

test=develop
上级 0bddb951
...@@ -501,6 +501,8 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor< ...@@ -501,6 +501,8 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor<
std::string flag = "--fraction_of_gpu_memory_to_use=" + std::string flag = "--fraction_of_gpu_memory_to_use=" +
std::to_string(fraction_of_gpu_memory); std::to_string(fraction_of_gpu_memory);
flags.push_back(flag); flags.push_back(flag);
// use auto growth strategy here.
flags.push_back("--allocator_strategy=auto_growth");
flags.push_back("--cudnn_deterministic=True"); flags.push_back("--cudnn_deterministic=True");
VLOG(3) << "set flag: " << flag; VLOG(3) << "set flag: " << flag;
framework::InitGflags(flags); framework::InitGflags(flags);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册