本文主要是介绍Ascend C算子开发(入门)—— 算子开发环境搭建,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
文章目录
- Ascend C算子开发(入门)—— 算子开发环境搭建
- 在CPU上部署开发环境
- 准备工作
- 步骤
- 配置环境变量
- 配置pip源
- pip安装依赖包
- 安装开发套件包
- 添加配置
- 验证是否安装成功
- 下载samples代码包
- 测试算子
Ascend C算子开发(入门)—— 算子开发环境搭建
安装包解读 Ascend-cann-功能-版本-平台.tar/run
- nnrt 推理
- amct 模型小型化工具
- communitysdk 社区算子开发工具包
- kernels 二进制安装包
- toolkit 开发套件
在CPU上部署开发环境
准备工作
一台安装ubuntu22.04桌面版(双系统/虚拟机均可)的PC机(或笔记本)
步骤
- 修改Ubuntu源
sudo passwd root # 修改root用户密码su # 登录root用户
# Password:... 输入密码cd /etc/apt
cp sources.list sources.list.bak
>sources.list
vim sources.list # 修改sources.list文件内容
------------------------------------------------
# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-updates main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-backports main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-security main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-security main restricted universe multiverse
# 预发布软件源,不建议启用
# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-proposed main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-proposed main restricted universe multiverse
------------------------------------------------
exit # 退出root用户
进入官网的CANN部分,点击立即下载中的社区版本。
找到最新版本的toolkit的软件安装包。具体的安装流程可以在文档中查看
进入文档后 ,点击CANN软件安装指南。
因为我们是Ubuntu系统,因此我们应该选择(Debian系列)的安装步骤指南。
sudo apt-get install -y gcc g++ make cmake libsqlite3-dev zlib1g-dev libssl-dev libffi-dev net-tools # 安装系统依赖
wget https://www.python.org/ftp/python/3.7.5/Python-3.7.5.tgz # 安装Python依赖
tar -zxvf Python-3.7.5.tgz
./configure --prefix=/usr/local/python3.7.5 --enable-loadable-sqlite-extensions --enable-shared
make -j16
sudo make install
配置环境变量
vim ~/.bashrc
-----------------------------------------------------------
export LD_LIBRARY_PATH=/usr/local/python3.7.5/lib:$LD_LIBRARY_PATH
export PATH=/usr/local/python3.7.5/bin:$PATH
-----------------------------------------------------------
source ~/.bashrc
配置pip源
cd # 切换到根目录
mkdir .pip
cd .pip
vim pip.conf
-----------------------------------------------------------
[global]
#以华为源为例,请根据实际情况进行替换。
index-url = https://mirrors.huaweicloud.com/repository/pypi/simple
trusted-host = mirrors.huaweicloud.com
timeout = 120
-----------------------------------------------------------
pip安装依赖包
pip3 install attrs cython numpy decorator sympy cffi pyyaml pathlib2 psutil protobuf scipy requests absl-py --user
安装开发套件包
cd
cd Downloads/
chmod +x Ascend-cann-toolkit_8.0.RC3.alpha002_linux-x86_64.run
./Ascend-cann-toolkit_8.0.RC3.alpha002_linux-x86_64.run --install
添加配置
vim ~/.bashrc
-----------------------------------------------------------
source /home/lby/Ascend/ascend-toolkit/set_env.sh
-----------------------------------------------------------
source ~/.bashrc
验证是否安装成功
下载samples代码包
在Gitee昇腾官网中的克隆/下载模块中找到命令
git clone https://gitee.com/ascend/samples.git
测试算子
cd samples/
cd operator/
cd AddCustomSample/
cd KernelLaunch/
cd AddKernelLaunchNeo/
bash run.sh -r cpu -v Ascend310P1
如果最终的测试结果是test pass
,说明算子能够正常执行,环境配置成功。
这篇关于Ascend C算子开发(入门)—— 算子开发环境搭建的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!