SNPE安装环境

参考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