设置 ML 框架

支持信息

此 Qualcomm® Neural Processing SDK 版本已经过验证，可与以下版本的 ML 训练框架配合使用：

框架	版本
TensorFlow	v2.10.1
TFLite	v2.3.0
TFLite	V1.13.1
ONNX	V1.12.0

将相应框架下载到主机上并进行安装

# Install tensorflow
pip install tensorflow==2.10.1

# Install tflite
pip install tflite==2.3.0

# Install PyTorch
pip install torch==1.13.1+cpu torchvision==0.14.1+cpu torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cpu

# Install Onnx
pip install onnx==1.12.0 onnxruntime==1.17.1 onnxsim==0.4.36

注意这个也比较久，请耐心等待

设置 SNPE SDK 环境

导入 SNPE SDK 提供的环境设置脚本，确保工作流程所需的所有工具和库均在 $PATH 中, 到此就可以进行转换了。

source ${SNPE_ROOT}/bin/envsetup.sh