一、环境配置
1、创建notebook并连接
使用ModelArts新建一个notebook,我这里镜像选择第一个,里面含有cann和Ascend910处理器,我这里环境只能使用ssh连接,创建一个密钥对,保存到C盘中的user/Administrator/目录下。
在网页中选择使用vscode接入,等待vscode打开后,选择密钥的位置等待连接成功。
2、查看npu型号
查看处理器的类型:npu-smi info。
从下图中可以看到这里是两块类型为昇腾910ProB的NPU,要记住这里的型号,后面在运行时用得到。
3、初始化环境
我这个notebook在创建时cmake的版本是3.12,低于要求的3.16,运行脚本使用更高版本的cmake,其他的环境都是符合条件的。下载文件后,放到work目录下,运行bash命令即可,等待脚本执行成功后,激活环境变量。然后使用cmake --version查看cmake的版本号如果是3.28则表示初始化激活环境变量成功。
# 1、初始化环境
bash init_env.sh
# 2、激活环境变量
source ~/.bashrc
二、运行add算子
使用git命令克隆官方的samples仓库,我这里放到work目录下,如果git不成功可以尝试多git几次,这里有可能因为网络原因git失败,我尝试git几次后成功clone下来。下载好仓库后进入Add算子中的目录,然后运行命