conda配置TensorFlow环境运行SNPE

约定不等于承诺〃 2022-12-30 07:59 126阅读 0赞

# 安装conda #

## conda简介 ##

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

###    安装conda ###

下载conda官网：[https://conda.io/miniconda.html][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

[https_conda.io_miniconda.html]: https://links.jianshu.com/go?to=https%3A%2F%2Fconda.io%2Fminiconda.html