update npu inference api (#9765)

2023-04-20 10:26:26 +08:00 · 2023-04-20 10:26:26 +08:00 · ce65745755
parent d496c8e8d1
commit ce65745755
1 changed files with 1 additions and 1 deletions
--- 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: