搭建Ascend环境¶
硬件与软件要求¶
| 类别 | 版本 | 说明 | |——–|——————–|———-| | Python | Python3.10 | 面向对象编程语言 | | CANN | 8.0.RC3.alpha003 | 昇腾异构计算架构 | | 系统镜像 | Ubuntu 22.04.5 LTS | 服务器操作系统 | | CPU架构 | aarch64 | CPU架构 | | NPU | Atlas 800I A2 | NPU型号 |
安装包准备¶
CANN社区版下载网址
下载好以下两个安装包
Ascend-cann-toolkit_8.0.RC3.alpha003_linux-aarch64.run
Ascend-cann-kernels-910b_8.0.RC3.alpha003_linux-aarch64.run
环境安装¶
参考官方文档
安装CANN
sudo chmod +x Ascend-cann-toolkit_8.0.RC3.alpha003_linux-aarch64.run
./Ascend-cann-toolkit_8.0.RC3.alpha003_linux-aarch64.run --install --force
安装Kernel
sudo chmod +x Ascend-cann-kernels-910b_8.0.RC3.alpha003_linux-aarch64.run
sudo ./Ascend-cann-kernels-910b_8.0.RC3.alpha003_linux-aarch64.run --install --install-for-all
更新环境 (如若指定了安装路径,请对应的修改此处的path)
sudo echo "source /usr/local/Ascend/ascend-toolkit/set_env.sh" >> /etc/profile
source /usr/local/Ascend/ascend-toolkit/set_env.sh
验证环境是否安装成功
git clone https://gitee.com/ascend/samples.git
cd samples/operator/AddCustomSample/KernelLaunch/AddKernelInvocation
bash run.sh -v Ascend910B4 -r npu
若成功则会如下图显示
