diff --git a/inferencer-fluid-gpu/include/fluid_gpu_engine.h b/inferencer-fluid-gpu/include/fluid_gpu_engine.h index 64fe14fba0e06adada1ab61e15198bcc10db5457..d2e014f4f58988da3f6edb8d0e6d2ff0b5a09daf 100644 --- a/inferencer-fluid-gpu/include/fluid_gpu_engine.h +++ b/inferencer-fluid-gpu/include/fluid_gpu_engine.h @@ -160,7 +160,7 @@ class FluidGpuNativeCore : public FluidFamilyCore { native_config.param_file = data_path + "/__params__"; native_config.prog_file = data_path + "/__model__"; native_config.use_gpu = true; - native_config.fraction_of_gpu_memory = 0.9; + native_config.fraction_of_gpu_memory = 0.01; native_config.device = FLAGS_gpuid; AutoLock lock(GlobalPaddleCreateMutex::instance()); _core = paddle::CreatePaddlePredictor_fluid_core = paddle::CreatePaddlePredictor(