diff --git a/deploy/utils/predictor.py b/deploy/utils/predictor.py
index 11f153071a0da0f82f035fe7389ad8f9f3bd8e6b..1c49c9dd5478932c655374fad541acbfc8952eeb 100644
--- a/deploy/utils/predictor.py
+++ b/deploy/utils/predictor.py
@@ -60,6 +60,7 @@ class Predictor(object):
                 precision_mode=Config.Precision.Half
                 if args.use_fp16 else Config.Precision.Float32,
                 max_batch_size=args.batch_size,
+                workspace_size=1 << 30,
                 min_subgraph_size=30)
 
         config.enable_memory_optim()