文章参考:https://www.hiascend.com/software/cann
我以Atlas 300I Duo卡安装CANN为例,310P芯片
安装依赖
openEuler系列操作系统包括:openEuler、CentOS、Kylin、BCLinux、BC-Linux-for-Euler、UOS20
1050e、UOS20 1020e、UOSV20、AntOS、CTyunOS、CULinux、Tlinux。
openEuler系列:
sudo yum makecache
配置最大线程数
训练场景下,OS的最大线程数可能不满足训练要求,需执行以下命令修改最大线程数为无限制。
配置环境变量,修改线程数为无限制,打开“/etc/profile”文件。
sudo vi /etc/profile
在文件的最后添加如下内容后保存退出。
ulimit -u unlimited
执行如下命令使环境变量生效。
source /etc/profile
sudo yum install -y gcc make net-tools python3 python3-devel python3-pip
检查系统是否安装满足版本要求的Python开发环境。
python3 --version
pip3 --version
CANN支持Python3.7.x至3.11.4版本,若未安装或者版本不满足请参考编译安装Python3.7.5操作。
执行如下命令安装所需的Python第三方库(如果使用root用户安装,请将命令中的–user删除)。
pip默认下载太慢,建议更换下载源:
sudo mkdir -p /etc/pip
sudo vim /etc/pip.conf
添加配置内容(以阿里云为例):[global]
index-url = https://mirrors.aliyun.com/pypi/simple/
trusted-host = mirrors.aliyun.com[install]
trusted-host = mirrors.aliyun.com
pip3 install attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py --user
这样就安装好了
安装CANN软件包
下载对应CANN软件包,只需要安装Toolkit开发套件包+Kernels算子包
Ascend-cann-toolkit_linux-.run
Ascend-cann-kernels-<chip_type>_linux-.run
(1)安装Toolkit开发套件包
chmod +x Ascend-cann-toolkit_<version>_linux-<arch>.run
./Ascend-cann-toolkit_<version>_linux-<arch>.run --check
出现如下回显信息,表示软件包校验成功。
Verifying archive integrity... 100% SHA256 checksums are OK. All good.
执行以上命令会打屏华为企业业务最终用户许可协议(EULA)的条款和条件,请输入Y或y同意协议,继续安装流程。
安装完成后,若显示如下信息,则说明软件安装成功:
source /usr/local/Ascend/ascend-toolkit/set_env.sh
安装后检查。执行如下命令查询CANN版本信息,查询结果与安装软件包的版本一致时,则验证安装成功。
进入软件包安装信息文件目录,请用户根据实际安装路径替换。表示CPU架构(aarch64或x86_64)。
cd /usr/local/Ascend/ascend-toolkit/latest/
执行命令,查看version字段提供的版本信息。
cat ascend_toolkit_install.info
(2)安装Kernels算子包
chmod +x Ascend-cann-kernels-<chip_type>_<version>_linux-<arch>.run
./Ascend-cann-kernels-<chip_type>_<version>_linux-<arch>.run --check
出现如下回显信息,表示软件包校验成功。
Verifying archive integrity... 100% SHA256 checksums are OK. All good.
请根据所需的文件,执行对应的命令安装软件包。
安装所有文件,包括单算子API执行(例如aclnn类API)动态库/静态库文件,以及kernel二进制文件,请执行如下命令:
./Ascend-cann-kernels-<chip_type>_<version>_linux-<arch>.run --devel
这样就安装成功了