diff --git a/tools/infer/utility.py b/tools/infer/utility.py
index 83451e949..959373cd7 100644
--- a/tools/infer/utility.py
+++ b/tools/infer/utility.py
@@ -246,7 +246,7 @@ def create_predictor(args, mode, logger):
                     logger.info("Please keep your paddlepaddle-gpu >= 2.3.0!")
 
         elif args.use_npu:
-            config.enable_npu()
+            config.enable_custom_device("npu")
         elif args.use_xpu:
             config.enable_xpu(10 * 1024 * 1024)
         else: