环境部署

在学习QuickStart或各类学习教程操作之前,请您先参考下面步骤完成基础环境搭建和源码下载,确保已安装NPU驱动、固件和CANN软件(Ascend-cann-toolkitAscend-cann-ops)等。

环境准备

本项目提供多种搭建昇腾环境的方式,请按需选择。

说明:本文提到的编译态和运行态含义如下,请根据实际情况选择。

  • 编译态:针对仅编译本项目不运行的场景,只需安装CANN toolkit包。
  • 运行态:针对运行本项目的场景(编译运行或纯运行),需安装驱动与固件、CANN toolkit包、CANN ops包。
安装方式 使用说明 使用场景
WebIDE 一站式开发平台,提供在线直接运行的昇腾环境,无需手动安装。
当前可提供单机算力,默认安装最新商发版CANN包
适用于没有昇腾设备的开发者。
Docker Docker镜像是一种高效部署方式,已预集成CANN包和必备依赖。
当前仅适用于Atlas A2系列产品,OS仅支持Ubuntu操作系统。默认安装最新商发版CANN包
适用有昇腾设备,需要快速搭建环境的开发者。
手动安装 - 适用有昇腾设备,想体验手动安装CANN包或体验最新master分支能力的开发者。

方式1:WebIDE环境

对于无昇腾设备的开发者,可直接使用WebIDE开发平台,即“算子一站式开发平台”,该平台为您提供在线可直接运行的昇腾环境,环境中已安装必备的驱动固件、软件包和依赖,无需手动安装。

说明:环境默认安装最新商发版CANN包,源码下载时注意与软件配套。更多关于开发平台的介绍请参考一站式算子开发工具平台操作指导

  1. 进入开源项目,单击“云开发”按钮,使用已认证过的华为云账号登录。若未注册或认证,请根据页面提示进行注册和认证。

    云平台
  2. 根据页面提示创建并启动云开发环境,单击“连接>WebIDE”进入算子一站式开发平台,开源项目的源码资源默认在/mnt/workspace目录下。

    云平台

方式2:Docker部署

对于有昇腾设备的开发者,若您想快速搭建昇腾环境,可使用Docker镜像部署。

说明

  • 镜像文件比较大,下载需要一定时间,请您耐心等待。关于docker命令的选项介绍可通过docker --help查询。
  • 环境默认安装最新商发版CANN包,源码下载时注意与软件配套。

1.安装驱动与固件(运行态依赖)

驱动与固件的下载和安装操作请参考《[CANN软件安装指南](https://www.hiascend.com/document/redirect/CannCommunityInstWizard)》中“准备软件包”和“安装NPU驱动和固件”章节。驱动与固件是运行态依赖,若仅编译算子,可以不安装。

2.下载镜像

  • 步骤1:以root用户登录宿主机。确保宿主机已安装Docker引擎(版本1.11.2及以上)。

  • 步骤2:从昇腾镜像仓库拉取已预集成CANN软件包及ops-cv所需依赖的镜像。命令如下,根据实际架构选择:

    # 示例:拉取ARM架构的CANN开发镜像
    docker pull --platform=arm64 swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-ops
    # 示例:拉取X86架构的CANN开发镜像
    docker pull --platform=amd64 swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-ops
    

3.运行Docker

拉取镜像后,需要以特定参数启动容器,以便容器内能访问宿主的昇腾设备。

docker 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
参数 说明 注意事项
--name cann_container 为容器指定名称,便于管理。 可自定义。
--device /dev/davinci0 核心:将宿主机的NPU设备卡映射到容器内,可指定映射多张NPU设备卡。 必须根据实际情况调整:davinci0对应系统中的第0张NPU卡。请先在宿主机执行 npu-smi info命令,根据输出显示的设备号(如NPU 0, NPU 1)来修改此编号。
--device /dev/davinci_manager 映射NPU设备管理接口。 -
--device /dev/devmm_svm 映射设备内存管理接口。 -
--device /dev/hisi_hdc 映射主机与设备间的通信接口。 -
-v /usr/local/dcmi:/usr/local/dcmi 挂载设备容器管理接口(DCMI)相关工具和库。 -
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi 挂载npu-smi工具。 使容器内可以直接运行此命令来查询NPU状态和性能信息。
-v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ 关键挂载:将宿主机的NPU驱动库映射到容器内。 -
-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info 挂载驱动版本信息文件。 -
-v /etc/ascend_install.info:/etc/ascend_install.info 挂载CANN软件安装信息文件。 -
-it -i(交互式)和 -t(分配伪终端)的组合参数。 -
swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:8.5.0-910b-ubuntu22.04-py3.10-ops 指定要运行的Docker镜像。 请确保此镜像名和标签(tag)与你通过docker pull拉取的镜像完全一致。
bash 容器启动后立即执行的命令。 -

方式3:手动安装

对于有昇腾设备的开发者,若您想手动搭建昇腾环境,请参考下述步骤。

前置依赖

请先确保编译环境的基础库依赖已安装,注意满足版本号要求。

  • python >= 3.7.0
  • gcc >= 7.3.0
  • cmake >= 3.16.0
  • pigz(可选,安装后可提升打包速度,建议版本 >= 2.4)
  • dos2unix
  • gawk
  • make
  • patch
  • googletest(仅执行UT时依赖,建议版本 release-1.11.0

上述依赖可通过项目根目录install_deps.sh一键安装,命令如下,若遇到不支持系统,请参考该文件自行适配。

bash install_deps.sh

安装完上述依赖后,可通过项目根目录requirements.txt继续安装python三方库依赖,命令如下。

pip3 install -r requirements.txt

软件安装

  • 场景1:体验master版本能力或基于master版本进行开发

    1. 安装驱动与固件(运行态依赖)

      下载和安装操作请参考《CANN软件安装指南》中“准备软件包”和“安装NPU驱动和固件”章节。驱动与固件是运行态依赖,若仅编译算子,可以不安装。

    2. 安装CANN包

      请单击下载链接,选择最新时间版本,并根据产品型号和环境架构下载对应包。安装命令如下,更多指导参考《CANN软件安装指南》。

      • 安装CANN toolkit包
      # 确保安装包具有可执行权限
      chmod +x Ascend-cann-toolkit_${cann_version}_linux-${arch}.run
      # 安装命令
      ./Ascend-cann-toolkit_${cann_version}_linux-${arch}.run --install --install-path=${install_path}
      
      • 安装CANN ops包(运行态依赖)

      ops包是运行态依赖,若仅编译算子,可不安装此包。

      # 确保安装包具有可执行权限
      chmod +x Ascend-cann-${soc_name}-ops_${cann_version}_linux-${arch}.run
      # 安装命令
      ./Ascend-cann-${soc_name}-ops_${cann_version}_linux-${arch}.run --install --install-path=${install_path}
          ```
      
      - \$\{cann\_version\}:表示CANN包版本号。
      - \$\{arch\}:表示CPU架构,如aarch64、x86_64。
      - \$\{soc\_name\}:表示NPU型号名称。
      - \$\{install\_path\}:表示指定安装路径,ops包需与toolkit包安装在相同路径,root用户默认安装在`/usr/local/Ascend`目录。
      
      
  • 场景2:体验已发布版本能力或基于已发布版本进行开发

    请访问CANN官网下载中心,选择发布版本(仅支持CANN 8.5.0及后续版本),并根据产品型号和环境架构下载对应包,最后参考网页提供的命令完成安装。

环境验证

安装完CANN包后,需验证环境和驱动是否正常。

  • 检查NPU设备

    # 运行npu-smi,若能正常显示设备信息,则驱动正常
    npu-smi info
    
  • 检查CANN版本

    # 查看CANN toolkit包版本信息(默认路径安装),WebIDE场景下将/usr/local替换为/home/developer
    cat /usr/local/Ascend/cann/${arch}-linux/ascend_toolkit_install.info
    # 查看CANN ops包版本信息(默认路径安装),WebIDE场景下将/usr/local替换为/home/developer
    cat /usr/local/Ascend/cann/${arch}-linux/ascend_ops_install.info
    

环境变量配置

按需选择合适的命令使环境变量生效。

# 默认路径安装,以root用户为例(非root用户,将/usr/local替换为${HOME})
source /usr/local/Ascend/cann/set_env.sh
# 指定路径安装
# source ${install_path}/cann/set_env.sh

源码下载

请根据CANN软件版本下载对应分支源码,${tag_version}表示分支标签名,分支标签与CANN版本配套关系参见release仓库

# 下载项目对应分支源码
git clone -b ${tag_version} https://gitcode.com/cann/ops-cv.git

对于WebIDE环境,已默认提供最新商发版本的项目源码,如需获取其他版本源码,也需通过上述命令下载源码。

注意

  • gitcode平台在使用HTTPS协议的时候要配置并使用个人访问令牌代替登录密码进行克隆,推送等操作。
  • 若您的编译环境无法访问网络,无法通过git指令下载代码,请先在联网环境中下载源码,再手动上传至目标环境。