AICPU
用法 |
python3 aicpu_generate_model.py --cores <core_num> -p <process_id> -t <npu_type> |
---|---|
参数名称 |
<core_num>:昇腾AI处理器AI Core的个数,默认为“2”。(预留参数,暂不使用) <process_id>:批量生成算子多进程调度的进程ID,默认值为“0”,无需设置。 <npu_type>:硬件形态,当前<npu_type>支持Atlas 200/300/500 推理产品、Atlas 推理系列产品和Atlas 800I A2推理产品,默认为“310”。如果无法确定具体的npu_type,则在安装昇腾AI处理器的服务器执行npu-smi info命令进行查询,将查询到的“Name”最后一位数字删除,即是npu_type的取值。 --help | -h:查询帮助信息。 |
说明 |
执行此命令,用户可以得到一组算子模型文件。 AICPU算子模型文件只需生成一次,会全部生成所有算法的算子。 |
父主题: 算子生成说明