本站点文档内容均翻译自code.visualstudio.com,仅供个人学习,如有差异请以官网为准。

手动设置模型转换环境

本文介绍了如何在不同的硬件目标(包括CPU、QNN、AMD NPU和Intel NPU)上手动设置Python环境以进行模型转换。

我们建议在Python 虚拟环境中Conda 环境中安装你的环境。

注意:如果您想在NPU设备上使用QNN EP进行推理,我们建议使用Python虚拟环境。原因是某些模型的推理可能需要ARM版本的Python,而Conda尚不支持创建ARM版本的Python虚拟环境。

设置 CPU 环境

提示:您可以直接设置NPU环境,例如设置QNN环境。在对模型进行推理时,它将回退到CPU EP。

或者,您可以选择设置一个CPU环境。

Python 版本:3.12.9

pip install requirements-CPU.txt

设置 QNN 环境

Python 版本:3.12.9

pip install requirements-QNN.txt

设置 AMD NPU 环境

Python 版本:3.10.9

pip install requirements-AMDNPU.txt

设置英特尔NPU环境

Python 版本:3.10.9

pip install requirements-IntelNPU.txt

目标项目的安装要求

导航到 模型项目文件夹.

使用pip安装requirements.txt中的包:pip install requirements.txt