昇腾社区首页
中文
注册
开发者
下载

安装后配置

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

安装运行所需依赖

运行业务时,依赖g++和以下Python第三方库,请执行如下命令安装:

  1. 安装g++(如果使用root用户安装,请将命令中的sudo删除)。
    Debian、Ubuntu、veLinux
    1
    sudo apt-get install -y g++
    
    openEuler、CentOS、Kylin、BCLinux、UOS V20、AntOS、AliOS、CTyunOS、CULinux、Tlinux、MTOS、vesselOS
    1
    sudo yum install -y gcc-c++
    
  2. 安装Python第三方库(如果使用root用户安装,请将命令中的--user删除)。
    1
    pip3 install attrs cython 'numpy>=1.19.2,<2.0' decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20.0 scipy requests absl-py --user
    

    以上示例命令会安装最新版本或指定版本的依赖,关于numpy等Python第三方库,需与Python版本配套使用,如遇报错,请参见运行依赖列表查看具体版本要求。

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

安装后验证

执行以下命令,若返回芯片型号,则验证CANN安装成功。

1
python3 -c "import acl;print(acl.get_soc_name())"

配置最大线程数(可选)

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

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