diff --git a/paddle/fluid/memory/allocation/allocator_facade.cc b/paddle/fluid/memory/allocation/allocator_facade.cc index cfc306dd35e66b75dc860d999de0238d5aa3b230..220b50b1cc392ba5734542277a466ad1aa279129 100644 --- a/paddle/fluid/memory/allocation/allocator_facade.cc +++ b/paddle/fluid/memory/allocation/allocator_facade.cc @@ -37,7 +37,7 @@ #endif DEFINE_int64( - gpu_allocator_retry_time, 100, + gpu_allocator_retry_time, 10000, "The retry time (milliseconds) when allocator fails " "to allocate memory. No retry if this value is not greater than 0"); diff --git a/python/paddle/fluid/__init__.py b/python/paddle/fluid/__init__.py index 218485f5f7545d91662439cb9af5089f8686d1c2..2395def84ea80e0877cf78bf6c7a9a8145594184 100644 --- a/python/paddle/fluid/__init__.py +++ b/python/paddle/fluid/__init__.py @@ -204,7 +204,7 @@ def __bootstrap__(): 'cudnn_exhaustive_search', 'selected_gpus', 'sync_nccl_allreduce', 'limit_of_tmp_allocation', 'times_excess_than_required_tmp_allocation', - 'cudnn_batchnorm_spatial_persistent' + 'cudnn_batchnorm_spatial_persistent', 'gpu_allocator_retry_time' ] core.init_gflags([sys.argv[0]] + ["--tryfromenv=" + ",".join(read_env_flags)])