CPU 和 GPU 版本模型部署

CPU 版本模型部署

模型部署在 CPU 设备时,将 部署设备 选择为 CPU

../../../_images/classification_cpu_deployment.png

GPU 版本模型部署

模型部署在 GPU 设备时,将 部署设备 选择为 GPU

只有训练模型与部署模型使用的设备 GPU 显卡型号相同时,需要将 导出模型类型 选择为 dlkmt 格式,否则选择 dlkmo 格式,以避免因导出和部署模型设备的显卡型号不同,而无法使用模型文件。

../../../_images/classification_gpu_deployment.png