diff --git a/deploy/utils/predictor.py b/deploy/utils/predictor.py
index 948b1859870d622ad370de1935775b0179a606b7..c13a40802ff9c3bdd55dd0ea47c4eec4294bad70 100644
--- a/deploy/utils/predictor.py
+++ b/deploy/utils/predictor.py
@@ -63,7 +63,7 @@ class Predictor(object):
         if args.get("use_gpu", False):
             config.enable_use_gpu(args.gpu_mem, 0)
         elif args.get("use_npu", False):
-            config.enable_npu()
+            config.enable_custom_device('npu')
         elif args.get("use_xpu", False):
             config.enable_xpu()
         else: