atlas200模型转换环境搭建

lijingle 深度学习 2023-11-17 08:44 389人围观

atlas200 环境搭建

Ubuntu环境下安装模型转换工具

1.首先安装anaconda，然后创建虚拟环境，

#创建虚拟环境
conda create --name cann python=3.8

2.激活虚拟环境

#激活虚拟环境
source activate cann

3.安装环境需要的依赖包

# 安装OS依赖
sudo apt-get install -y gcc g++ make cmake zlib1g zlib1g-dev openssl libsqlite3-dev libssl-dev libffi-dev unzip pciutils net-tools libblas-dev gfortran libblas3 libopenblas-dev

# 安装Python3依赖
apt-get install -y python3-pip
pip3 install --upgrade pip
pip3 install attrs cython numpy decorator sympy cffi pyyaml pathlib2 psutil protobuf scipy requests absl-py

4.下载atlas工具包tookit

进入网站

https://www.hiascend.com/developer/download/community?module=pt+tf+cann

选择合适的产品，否则转换成功也加载不上。

如图这里选择开发者套件，以及atlas200 型号3000

下载cann tookit套件

如果在电脑上就下载x86_64，如果开发板则是aarch64版本

# 在/home目录下执行如下命令安装Toolkit
./Ascend-cann-toolkit_7.0.RC1_linux-x86_64.run --install 

# 执行如下命令配置环境变量。若需要设置环境变量永久生效，可在~/.bashrc文件最后一行后面添加以下命令，执行source ~/.bashrc命令
source /usr/local/Ascend/ascend-toolkit/set_env.sh

安装完成后添加环境变量：

如果root用户下安装：

source  /usr/local/Ascend/ascend-toolkit/set_env.sh
#若开发套件包Ascend-cann-toolkit在非昇腾设备上安装，则如下环境变量必须执行，用于设置动态链接库所在路径，否则无需执行
export LD_LIBRARY_PATH=Ascend-cann-toolkit安装目录/ascend-toolkit/latest/<arch>-linux/devlib

非root用户下

. ${HOME}/Ascend/ascend-toolkit/set_env.sh
#若开发套件包Ascend-cann-toolkit在非昇腾设备上安装，则如下环境变量必须执行，用于设置动态链接库所在路径，否则无需执行
export LD_LIBRARY_PATH=Ascend-cann-toolkit安装目录/ascend-toolkit/latest/<arch>-linux/devlib

我的安装在非root用户下
在~/.bashrc 文件下添加如下内容
source /home/lijingle/Ascend/ascend-toolkit/set_env.sh
export LD_LIBRARY_PATH=/home/lijingle/Ascend/ascend-toolkit/latest/x86_64-linux/devlib/:$LD_LIBRARY_PATH