算子开发快速入门:基于ops-rand仓
算子开发快速入门基于ops-rand仓【免费下载链接】ops-randops-rand是CANN Compute Architecture for Neural Networks算子库中提供的随机数生成库。项目地址: https://gitcode.com/cann/ops-rand本指南旨在帮助你快速上手基于CANN和ops-rand算子仓的使用最简化地完成环境安装、编译部署及算子运行。一、环境安装1. 有环境场景Docker安装Docker安装环境以Atlas A2产品910B为例。前提条件Docker环境宿主机已安装Docker引擎版本1.11.2及以上。驱动与固件宿主机已安装昇腾NPU的驱动与固件Ascend HDK 24.1.0版本以上。安装指导详见《CANN 软件安装指南》。注意使用npu-smi info查看对应的驱动与固件版本。下载镜像拉取已预集成CANN软件包及ops-rand所需依赖的镜像。操作步骤以root用户登录宿主机。执行拉取命令请根据你的宿主机架构选择ARM架构docker pull --platformarm64 swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-opsX86架构docker pull --platformamd64 swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-ops注意正常网速下镜像下载时间约为5-10分钟。Docker运行请根据以下命令运行dockerdocker run --name cann_container --device /dev/davinci0 --device /dev/davinci_manager --device /dev/devmm_svm --device /dev/hisi_hdc -v /usr/local/dcmi:/usr/local/dcmi -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info -v /etc/ascend_install.info:/etc/ascend_install.info -it swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-ops bash检查环境进入容器后验证环境和驱动是否正常。检查NPU设备npu-smi info检查CANN安装cat /usr/local/Ascend/ascend-toolkit/latest/opp/version.info2. 无环境场景WebIDE开发建设中对于无环境的用户提供WebIDE开发方式目前本方式正在建设中。二、编译部署1. 拉取ops-rand仓库代码git clone https://gitcode.com/cann/ops-rand.git cd ops-rand2. 编译ops-rand算子包bash build.sh --pkg --socascend950若提示如下信息说明编译成功。Self-extractable archive cann-950-ops-rand_9.0.0_linux-x86_64.run successfully created. Build package success: build_out/cann-950-ops-rand_9.0.0_linux-x86_64.run编译成功后run包存放于项目根目录的build_out目录下。3. 安装ops-rand算子包./build_out/cann-${soc_name}-ops-rand_${cann_version}_linux-${arch}.run --full --install-path${install_path}${soc_name}表示NPU型号名称如950。${cann_version}表示CANN版本号如9.0.0。${arch}表示CPU架构如aarch64、x86_64。${install_path}表示指定安装路径需要与toolkit包安装在相同路径默认安装在/usr/local/Ascend目录。ops-rand安装在${install_path}/cann路径中。4. 配置环境变量source ${ASCEND_HOME_PATH}/cann/set_env.bash5. 运行测试bash build.sh --run三、更多帮助CANN 开发文档Ascend C API 参考【免费下载链接】ops-randops-rand是CANN Compute Architecture for Neural Networks算子库中提供的随机数生成库。项目地址: https://gitcode.com/cann/ops-rand创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考