diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index 8d66f8671815c53d00476033f772cb58069606c8..1d07f62f206ca2c250e340921e862bbccd5382b2 100755
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -101,6 +101,7 @@ def create_predictor(args, mode):
         config.disable_gpu()
         config.set_cpu_math_library_num_threads(6)
         if args.enable_mkldnn:
+            config.set_mkldnn_cache_capacity(10)
             config.enable_mkldnn()
 
     #config.enable_memory_optim()