环境部署#

方式1：CANNLab#

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

说明：环境默认安装最新商发版CANN包，源码下载时注意与软件配套。更多关于开发平台的介绍请参考CANNLab指导。

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

2. 根据页面提示信息创建并启动云开发环境，单击”连接 > WebIDE”进入一站式开发平台。

方式2：Docker部署#

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

说明：

• 镜像文件比较大，下载需要一定时间，请您耐心等待。关于docker命令的选项介绍可通过docker --help查询。

• 环境默认安装最新商发版CANN包，源码下载时注意与软件配套。

1. 安装驱动与固件（运行态依赖）

   驱动与固件是运行态依赖，若仅编译PyPTO，可不安装。使用npu-smi info检查是否有NPU相关信息，若没有，请参考《CANN快速安装》完成驱动与固件安装。

2. 下载镜像

   • 步骤1：以root用户登录宿主机。确保宿主机已安装Docker引擎（版本1.11.2及以上），使用docker --version检查Docker版本，若没有，请参考Docker官方安装指南。

   • 步骤2：从昇腾镜像仓库拉取已预集成CANN软件包及PyPTO运行所需依赖的镜像。

     示例如下，请自行替换CANN版本号、芯片系列、操作系统、python版本信息。

     # 以cann:9.1.0-beta.1版本为例
     docker pull swr.cn-south-1.myhuaweicloud.com/ascendhub/cann:9.1.0-beta.1-910b-openeuler24.03-py3.12-devel
     

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:9.1.0-beta.1-910b-openeuler24.03-py3.12-devel 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:9.1.0-beta.1-910b-openeuler24.03-py3.12-devel	指定要运行的Docker镜像。	请确保此镜像名和标签（tag）与你通过docker pull拉取的镜像完全一致。
   bash	容器启动后立即执行的命令。	-

4. 安装PyPTO依赖

   进入容器后，请参考手动安装 - PyPTO依赖完成Python依赖及其它编译依赖的安装。

方式3：手动安装#

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

PyPTO Toolkit插件#

如需体验计算图和泳道图的查看能力，请安装PyPTO Toolkit插件： 具体使用文档参考 PyPTO Toolkit 文档

MPI依赖#

PyPTO的分布式用例依赖MPI，推荐版本 >= 3.2.1。

方式一：通过系统包管理器安装

# Ubuntu/Debian系统
apt-get update && apt-get install -y mpich

# CentOS/RHEL系统
yum install -y mpich

方式二：通过源码编译安装

# 以3.2.1版本为例
version='3.2.1'
wget https://www.mpich.org/static/downloads/${version}/mpich-${version}.tar.gz
tar -xzf mpich-${version}.tar.gz
cd mpich-${version}
./configure --prefix=/usr/local/mpich --disable-fortran
make && make install

安装完成后设置环境变量：

export MPI_HOME=/usr/local/mpich
export PATH=${MPI_HOME}/bin:${PATH}