fix jetson problem (#30939)

revert-31068-fix_conv3d_windows
Wilber 4 years ago committed by GitHub
parent 01ccfbcde9
commit 39aeaa160e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save