昇腾社区首页
中文
注册

安装后操作

安装软件包后,请根据业务要求参考本节内容进行配置。

安装运行所需依赖

  1. 安装g++。
    Debian系列:
    1
    sudo apt-get install -y g++
    
    openEuler系列:
    1
    sudo yum install -y gcc-c++
    
  2. 安装Python第三方库(如果使用root用户安装,请将命令中的--user删除)。
    1
    pip3 install attrs cython 'numpy>=1.19.2,<=1.24.0' decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20.0 scipy requests absl-py --user
    

    以上命令会安装最新版本或指定版本的依赖,关于Python第三方库、glibc版本要求请参见CANN依赖列表

    若安装过程中提示pip源不可以用,请参见配置pip源,完成后再执行安装命令。

运行推理样例

可通过运行一个简单的推理样例,来验证CANN安装的完整性。

单击Link获取快速入门样例,并按README.md中的指导下载样例、准备模型、准备测试图片、编译及运行应用,如果CANN软件安装配置正确,则输出的结果会类似于样例中示例信息。

1
2
3
4
5
top 1: index[162] value[0.954676]
top 2: index[161] value[0.033442]
top 3: index[166] value[0.006534]
top 4: index[167] value[0.004561]
top 5: index[163] value[0.000315]

配置最大线程数(可选)

训练场景下,OS的最大线程数可能不满足训练要求,以root用户执行以下命令修改最大线程数为无限制。

  1. 配置环境变量,修改线程数为无限制,打开“/etc/profile”文件。
    1
    vi /etc/profile
    
  2. 添加如下内容后保存退出。
    1
    ulimit -u unlimited
    
  3. 执行如下命令使环境变量生效。
    1
    source /etc/profile
    

指定标准库头文件路径(可选)

在openEuler 24.03场景下,由于该OS默认的gcc编译器所在路径与常见OS上的编译器路径有差异,在算子开发时,需要通过手动配置CPLUS_INCLUDE_PATH环境变量指定标准库头文件的路径,使毕昇编译器能查找到gcc编译器的头文件。

  • aarch64架构:
    1
    export CPLUS_INCLUDE_PATH=/usr/include/c++/12:/usr/include/c++/12/backward:/usr/include/c++/12/aarch64-openEuler-linux/:$CPLUS_INCLUDE_PATH
    
  • x86_64
    1
    export CPLUS_INCLUDE_PATH=/usr/include/c++/12:/usr/include/c++/12/backward:/usr/include/c++/12/x86_64-openEuler-linux/:$CPLUS_INCLUDE_PATH
    

手动配置环境变量(可选)

针对ToolkitNNAE,除了CANN提供进程级环境变量设置脚本set_env.sh外,用户还可以通过手动配置环境变量。

Toolkit为例,可执行如下示例命令配置环境变量:

export ASCEND_TOOLKIT_HOME=/usr/local/Ascend/ascend-toolkit/latest
export PYTHONPATH=${ASCEND_TOOLKIT_HOME}/python/site-packages:${ASCEND_TOOLKIT_HOME}/opp/built-in/op_impl/ai_core/tbe:$PYTHONPATH
export LD_LIBRARY_PATH=${ASCEND_TOOLKIT_HOME}/lib64:$LD_LIBRARY_PATH
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的默认安装路径为例,root用户:“/usr/local/Ascend/ascend-toolkit/latest”,非root用户:“${HOME}/Ascend/ascend-toolkit/latest”。