|
|
|
@ -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: ";
|
|
|
|
|