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

test=develop
paddle_tiny_install
Zhaolong Xing 6 years ago committed by GitHub
parent 0bddb951c2
commit b39c011637
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -501,6 +501,8 @@ std::unique_ptr<PaddlePredictor> CreatePaddlePredictor<
std::string flag = "--fraction_of_gpu_memory_to_use=" +
std::to_string(fraction_of_gpu_memory);
flags.push_back(flag);
// use auto growth strategy here.
flags.push_back("--allocator_strategy=auto_growth");
flags.push_back("--cudnn_deterministic=True");
VLOG(3) << "set flag: " << flag;
framework::InitGflags(flags);

Loading…
Cancel
Save