参考SNPE官方文档的步骤。本文以安装pytorch为例记录安装过程,其他类似。
一 准备工作
1)ubuntu18。04;
2)不同框架的安装步骤;
3)python3的环境;
二 SNPE的环境以及依赖
注意安装python的各种package;
三 pytorch环境的安装;
3.1 pytorch setup;
根据pytorch官网不同版本的命令行运行,生成不同的版本;
pip3 install torch==1.8.2+cpu torchvision==0.9.2+cpu torchaudio==0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html $ python -c "import torch" >>> import torch >>> torch.__path__ ['/home/worker/.local/lib/python3.6/site-packages/torch'] export ONNX_DIR=/home/worker/.local/lib/python3.6/site-packages/torch
注意,python -c "import torch"中是小写的字母c;
3.2 Get TVM
git clone --recursive https://github.com/apache/tvm tvm cd tvm git checkout 0f4c0654 git submodule update --init mkdir build cp cmake/config.cmake build
这个步骤需要FQ,否则很慢;
3.3 LLVM config
下载某个版本的LLVM,需要FQ,注意,版本需要大于4.0,但是测试13.0不可以,10.0版本可以;
在TVM的build/config.cmake中配置USE_LLVM选项:
set(USE_LLVM /path/to/your/llvm/bin/llvm-config)
3.4 Build TVM
3.5 test pytorch
四 问题总结
注意,每个环境是在一个新的/干净的虚拟环境中安装的,也可以在本机安装;可以使用conda,也可以docker;
本文使用的是docker,不同的框架环境要安装在不同的虚拟环境中,也就是不同的docker容器安装的是不同框架的环境。
参考
1
https://developer.qualcomm.com/software/qualcomm-neural-processing-sdk/getting-started
2。
https://developer.qualcomm.com/sites/default/files/docs/snpe/setup.html