下载
中文
注册

安装包相关

描述CANN软件安装完成后,系统通过“set_env.sh”脚本默认配置的程序编译、执行相关的基础环境变量,以及后续程序编译执行时,需要手工配置的跟安装包相关的环境变量。

安装后自动配置环境变量

CANN软件安装完成后,默认会自动配置程序编译、执行所依赖的基础环境变量,但用户进程结束后相关环境变量自动失效,您也可以通过如下环境变量脚本一键式配置。

# root用户安装时环境变量配置示例:
# 安装toolkit包时
source /usr/local/Ascend/ascend-toolkit/set_env.sh
# 安装nnrt包时
source /usr/local/Ascend/nnrt/set_env.sh    
# 安装nnae包时
source /usr/local/Ascend/nnae/set_env.sh   
#安装tfplugin包时
source /usr/local/Ascend/tfplugin/set_env.sh
# 非root用户安装时环境变量配置示例:
# 安装toolkit时
source ${HOME}/Ascend/ascend-toolkit/set_env.sh 
# 安装nnrt包时
source ${HOME}/Ascend/nnrt/set_env.sh    
# 安装nnae包时
source ${HOME}/Ascend/nnae/set_env.sh
# 安装tfplugin包时
source ${HOME}/Ascend/tfplugin/set_env.sh

用户也可以通过修改~/.bashrc文件方式设置永久环境变量,操作如下:

  1. 以运行用户在任意目录下执行vi ~/.bashrc命令,打开.bashrc文件,在文件最后一行后面添加上述内容。
  2. 执行:wq!命令保存文件并退出。
  3. 执行source ~/.bashrc命令使其立即生效。

详细环境变量说明如下表所示。

表1 环境变量说明

软件包

环境变量名称

环境变量说明

配置示例

开发套件包(toolkit)

LD_LIBRARY_PATH

动态库的查找路径。

export LD_LIBRARY_PATH=/usr/local/Ascend/driver/lib64:/usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/driver:$LD_LIBRARY_PATH
export ASCEND_TOOLKIT_HOME=/home/HwHiAiUser/Ascend/ascend-toolkit/latest
export LD_LIBRARY_PATH=${ASCEND_TOOLKIT_HOME}/lib64:${ASCEND_TOOLKIT_HOME}/lib64/plugin/opskernel:${ASCEND_TOOLKIT_HOME}/lib64/plugin/nnengine:$LD_LIBRARY_PATH
export PYTHONPATH=${ASCEND_TOOLKIT_HOME}/python/site-packages:${ASCEND_TOOLKIT_HOME}/opp/built-in/op_impl/ai_core/tbe:$PYTHONPATH
export PATH=${ASCEND_TOOLKIT_HOME}/bin:${ASCEND_TOOLKIT_HOME}/compiler/ccec_compiler/bin:$PATH
export ASCEND_AICPU_PATH=${ASCEND_TOOLKIT_HOME}
export ASCEND_OPP_PATH=${ASCEND_TOOLKIT_HOME}/opp
export TOOLCHAIN_HOME=${ASCEND_TOOLKIT_HOME}/toolkit
export ASCEND_HOME_PATH=${ASCEND_TOOLKIT_HOME}

ASCEND_TOOLKIT_HOME

CANN-toolkit软件包安装后文件存储路径。

PYTHONPATH

Python搜索路径

PATH

可执行文件查找路径。

ASCEND_AICPU_PATH

AI CPU组件的安装路径。

ASCEND_OPP_PATH

算子库根目录。

TOOLCHAIN_HOME

工具链安装路径。

ASCEND_HOME_PATH

同ASCEND_TOOLKIT_HOME,代表CANN-toolkit软件安装后文件存储路径。

离线推理引擎包(nnrt)

ASCEND_NNRT_HOME

nnrt软件包安装后文件存储路径。

export LD_LIBRARY_PATH=/usr/local/Ascend/driver/lib64:/usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/driver:$LD_LIBRARY_PATH
export ASCEND_NNRT_HOME=/home/HwHiAiUser/Ascend/nnrt/latest
export LD_LIBRARY_PATH=${ASCEND_NNRT_HOME}/lib64:$LD_LIBRARY_PATH
export PYTHONPATH=${ASCEND_NNRT_HOME}/python/site-packages:$PYTHONPATH
export ASCEND_AICPU_PATH=${ASCEND_NNRT_HOME}
export ASCEND_OPP_PATH=${ASCEND_NNRT_HOME}/opp

LD_LIBRARY_PATH

动态库的查找路径。

PYTHONPATH

Python搜索路径

ASCEND_AICPU_PATH

AI CPU组件的安装路径。

ASCEND_OPP_PATH

算子库根目录。

深度学习引擎包(nnae)

ASCEND_NNAE_HOME

nnae软件包安装后文件存储路径。

export LD_LIBRARY_PATH=/usr/local/Ascend/driver/lib64:/usr/local/Ascend/driver/lib64/common:/usr/local/Ascend/driver/lib64/driver:$LD_LIBRARY_PATH
export ASCEND_NNAE_HOME=/home/HwHiAiUser/Ascend/nnae/latest
export LD_LIBRARY_PATH=${ASCEND_NNAE_HOME}/lib64:${ASCEND_NNAE_HOME}/lib64/plugin/opskernel:${ASCEND_NNAE_HOME}/lib64/plugin/nnengine:$LD_LIBRARY_PATH
export PYTHONPATH=${ASCEND_NNAE_HOME}/python/site-packages:${ASCEND_NNAE_HOME}/opp/built-in/op_impl/ai_core/tbe:$PYTHONPATH
export PATH=${ASCEND_NNAE_HOME}/bin:${ASCEND_NNAE_HOME}/compiler/ccec_compiler/bin:$PATH
export ASCEND_AICPU_PATH=${ASCEND_NNAE_HOME}
export ASCEND_OPP_PATH=${ASCEND_NNAE_HOME}/opp
export ASCEND_HOME_PATH=${ASCEND_NNAE_HOME}

LD_LIBRARY_PATH

动态库的查找路径。

PYTHONPATH

Python搜索路径

PATH

可执行文件查找路径。

ASCEND_AICPU_PATH

AI CPU组件的安装路径。

ASCEND_OPP_PATH

算子库根目录。

ASCEND_HOME_PATH

同ASCEND_NNAE_HOME,代表nnae软件安装后文件存储路径。

TensorFlow框架插件包(tfplugin)

ASCEND_PLUGIN_HOME

框架包安装后文件存储路径。

export ASCEND_PLUGIN_HOME=/usr/local/Ascend/tfplugin/latest
export PYTHONPATH=${ASCEND_PLUGIN_HOME}/python/site-packages:$PYTHONPATH

PYTHONPATH

Python搜索路径

表2 环境变量说明

软件包

环境变量名称

环境变量说明

配置示例

Runtime

LD_LIBRARY_PATH

动态库的查找路径

# Runtime使用到的so文件存在路径
# 如环境中存在Driver,则配置如下。
export LD_LIBRARY_PATH=${install_path}/latest/runtime/lib64:$LD_LIBRARY_PATH
# 如环境中不存在Driver,则配置如下。
export LD_LIBRARY_PATH=${install_path}/latest/runtime/lib64:$LD_LIBRARY_PATH:${install_path}/latest/runtime/lib64/stub

Compiler

PYTHONPATH

Python搜索路径

# Compiler使用的python路径
export PYTHONPATH=${install_path}/latest/compiler/python/site-packages:$PYTHONPATH

# Compiler可执行程序的查找路径。如果用户环境存在多个python3版本,则指定使用python3.7.5版本,python3.7.5安装路径请根据实际情况进行替换;同时设置ATC工具可执行文件所在路径
export PATH=${install_path}/latest/compiler/ccec_compiler/bin:${install_path}/latest/compiler/bin:$PATH

# Compiler的库文件存放路径
export LD_LIBRARY_PATH=${install_path}/latest/compiler/lib64:${install_path}/latest/compiler/lib64/plugin/opskernel:${install_path}/latest/compiler/lib64/plugin/nnengine:$LD_LIBRARY_PATH:${install_path}/latest/compiler/lib64/stub

PATH

可执行程序的查找路径

LD_LIBRARY_PATH

动态库的查找路径。

HCCL

PYTHONPATH

Python搜索路径

# HCCL使用的python路径
export PYTHONPATH=${install_path}/latest/hccl/python/site-packages:$PYTHONPATH

# HCCL可执行程序的查找路径。如果用户环境存在多个python3版本,则指定使用python3.7.5版本,python3.7.5安装路径请根据实际情况进行替换;同时设置ATC工具可执行文件所在路径
export PATH=${install_path}/latest/hccl/bin:$PATH

# HCCL的库文件存在路径
export LD_LIBRARY_PATH=${install_path}/latest/hccl/lib64:$LD_LIBRARY_PATH:${install_path}/latest/hccl/lib64/plugin/opskernel

PATH

可执行程序的查找路径

LD_LIBRARY_PATH

动态库的查找路径。

Toolkit

TOOLCHAIN_HOME

工具链安装路径。

# 工具链安装路径
export TOOLCHAIN_HOME=${install_path}/latest/toolkit

# Toolkit使用的python路径
export PYTHONPATH=${install_path}/latest/toolkit/python/site-packages:$PYTHONPATH

# Toolkit的库文件存在路径
export LD_LIBRARY_PATH=${install_path}/latest/toolkit/tools/aml/lib64:$LD_LIBRARY_PATH

# Toolkit可执行程序的查找路径 
export PATH=${install_path}/latest/toolkit/python/site-packages/bin/:${install_path}/latest//toolkit/tools/profiler/bin/:${install_path}/latest/toolkit/tools/ascendc_tools/:${install_path}/latest/toolkit/tools/aml/lib64/:${install_path}/latest/toolkit/tools/ccec_compiler/bin:$PATH

PYTHONPATH

Python搜索路径

LD_LIBRARY_PATH

动态库的查找路径。

PATH

可执行程序的查找路径

OPP

ASCEND_OPP_PATH

算子库根目录。

# 算子库根目录
export ASCEND_OPP_PATH=${install_path}/latest/opp

#AI CPU组件的安装路径
export ASCEND_AICPU_PATH=${install_path}/latest

# OPP使用的python路径
export PYTHONPATH=${install_path}/latest/opp/built-in/op_impl/ai_core/tbe/:$PYTHONPATH

ASCEND_AICPU_PATH

AI CPU组件的安装路径。

PYTHONPATH

Python搜索路径

如果安装了AOE,还有如下环境变量。

AOE

LD_LIBRARY_PATH

动态库的查找路径。

# AOE的库文件存在路径
export LD_LIBRARY_PATH=${install_path}/latest/tools/aoe/lib64:$LD_LIBRARY_PATH

# AOE可执行程序的查找路径
export PATH=${install_path}/latest/tools/aoe:$PATH

PATH

可执行程序的查找路径

手工配置环境变量

程序编译执行过程中,可能会需要配置跟软件包安装相关的环境变量。下面描述CANN提供的不同场景下需要手工配置的环境变量。

表3 手工配置环境变量列表

软件包

环境变量名称

环境变量说明

配置示例

自定义算子安装包

ASCEND_CUSTOM_OPP_PATH

用户自定义算子包安装路径。

支持配置多个路径,以英文冒号分隔,排在最前面的优先级最高。

export ASCEND_CUSTOM_OPP_PATH=${install_path}/ascend-toolkit/latest/opp/vendor_name1:${install_path}/ascend-toolkit/latest/opp/vendor_name2