conda配置TensorFlow环境运行SNPE

安装conda

 

conda简介

Conda 是一个开源的软件包管理系统和环境管理系统，用于安装多个版本的软件包及其依赖关系，并在它们之间轻松切换。 Conda 是为 Python 程序创建的，适用于 Linux，OS X 和Windows，也可以打包和分发其他软件。

安装conda

下载conda官网：https://conda.io/miniconda.html

bash Anaconda3-2020.07-Linux-x86_64.sh

安装完成后需要打开新的terminal窗口，因为conda已经自动在~/.bashrc里面配置好了环境：

# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/home/hui/data/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/home/hui/data/anaconda3/etc/profile.d/conda.sh" ]; then
        . "/home/hui/data/anaconda3/etc/profile.d/conda.sh"
    else
        export PATH="/home/hui/data/anaconda3/bin:$PATH"
    fi
fi
unset __conda_setup
# <<< conda initialize <<<

安装配置tensorflow环境

1. 创建tensorflow的env，命名为tensorflow

conda create -n tensorflow python=3.5

2. 激活创建的tensorflow ENV

conda activate tensorflow

3. 安装tensorflow=1.14，因为snpe工具不能安装TensorFlow 2.0以上版本，所以指定为1.14

conda install tensorflow=1.14

安装报错，发现和snpe需要的python3.5冲突，这时候使用pip安装居然是OK的。
 

Specifications:

  - tensorflow=1.14 -> python[version='2.7.*|3.6.*|3.7.*']
  - tensorflow=1.14 -> python[version='>=2.7,<2.8.0a0|>=3.6,<3.7.0a0|>=3.7,<3.8.0a0']

Your python: python=3.5

4. 在conda下使用pip安装tensorflow1.14

pip install tensorflow=1.14

5. 完成后查看env info

conda  info 

  active environment : tensorflow
  active env location : /home/hui/data/anaconda3/envs/tensorflow

6. 设置snpe工具运行需要的环境变量

source bin/envsetup.sh -t /home/hui/data/anaconda3/envs/tensorflow

export PYTHONPATH=$PYTHONPATH:/home/hui/tools/snpe-1.36.0.746/lib/python/snpe/dlc_utils/:/home/hui/tools/snpe-1.36.0.746/benchmarks/common_utils:/home/hui/tools/snpe-1.36.0.746/benchmarks/snpebm:/home/hui/tools/snpe-1.36.0.746/benchmarks

7. 执行pb to dlc

snpe-tensorflow-to-dlc --input_network ./step.pb --output_path step.dlc -d input_node "1,384,384,3" --out_node "output_node" --allow_unconsumed_nodes

8. 全部完成后

conda deactivate

9. 下次使用时，再次激活，然后设置环境变量就可以使用SNPE的相关tools了。

$ conda activate tensorflow

$ source bin/envsetup.sh -t /home/hui/data/anaconda3/envs/tensorflow

$ export PYTHONPATH=$PYTHONPATH:/home/hui/disk1t/snpe/snpe-1.44.1/snpe-1.44.1.2358/lib/python/snpe/dlc_utils/:/home/hui/disk1t/snpe/snpe-1.44.1/snpe-1.44.1.2358/benchmarks/common_utils:/home/hui/disk1t/snpe/snpe-1.44.1/snpe-1.44.1.2358/benchmarks/snpebm:/home/hui/disk1t/snpe/snpe-1.44.1/snpe-1.44.1.2358/benchmarks