参加你附近的 ,了解 VS Code 中的 AI 辅助开发。

手动设置模型转换环境

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

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

注意:如果您想使用 QNN EP 在 NPU 设备上进行推理,我们建议使用 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

设置 Intel NPU 环境

Python 版本:3.10.9

pip install requirements-IntelNPU.txt

安装目标项目的要求

导航到模型项目文件夹

pip install requirements.txt