在创建config时,可以通过设置paddle_infer.Config的参数来选择cpu。具体来说,可以通过设置use_gpu=False来选择cpu。代码如下所示:

config = paddle_infer.Config(args.model_file, args.params_file) config.disable_gpu() # 选择cpu

另外,如果您的电脑有多个cpu,您也可以通过设置cpu_math_library_num_threads来控制使用的cpu数量。代码如下所示:

config.set_cpu_math_library_num_threads(4) # 使用4个cpu

import argparseimport numpy as np# 引用 paddle inference 推理库import paddleinference as paddle_inferdef main args = parse_args # 创建 config config = paddle_inferConfigargsmodel_file argsparams_fil

原文地址: https://www.cveoy.top/t/topic/haX2 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录