快速安装昇腾环境 — 昇腾开源 文档 (original) (raw)

快速安装昇腾环境

跟随指导，在您的机器上快速安装昇腾环境。

1. 系统要求

1.1 前置检查

确认昇腾AI处理器已经安装妥当

lspci | grep 'Processing accelerators'

确认操作系统架构及版本

uname -m && cat /etc/*release

确认Python版本

1.2 软件要求

2. 环境安装

根据您的需求，选择合适的软件包版本：

安装方式

操作系统

CPU架构

NPU型号

昇腾套件版本

NPU型号

Atlas 300T A2

Atlas 300I Duo

2.1 安装驱动

2.1.1 安装依赖

sudo apt-get install -y gcc make net-tools python3 python3-dev python3-pip

sudo yum install -y gcc make net-tools python3 python3-devel python3-pip

2.1.2 创建驱动运行用户

备注

请使用命令 id HwHiAiUser 查看用户是否存在，若存在请跳过此步骤

sudo groupadd HwHiAiUser
sudo useradd -g HwHiAiUser -d /home/HwHiAiUser -m HwHiAiUser -s /bin/bash
sudo usermod -aG HwHiAiUser $USER

2.1.3 下载并安装

确认您的驱动是否安装成功，可以通过以下命令验证：npu-smi info，若出现以下回显信息，说明驱动安装成功。

+-------------------------------------------------------------------------------------------+ | npu-smi 23.0.2 Version: 23.0.2 | +----------------------+---------------+----------------------------------------------------+ | NPU Name | Health | Power(W) Temp(C) Hugepages-Usage(page)| | Chip | Bus-Id | AICore(%) Memory-Usage(MB) HBM-Usage(MB) | +======================+===============+====================================================+ | 0 xxx | OK | 0.0 40 0 / 0 | | 0 | 0000:C1:00.0 | 0 882 / 15169 0 / 32768 | +======================+===============+====================================================+

2.2 安装固件

安装固件后，若系统出现如下关键回显信息，表示固件安装成功。

Firmware package installed successfully!

2.3 安装CANN

2.3.1 安装python依赖

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py

2.3.2 下载并安装

安装CANN-toolkit后，若系统出现以下关键回显信息，表示CANN-toolkit安装成功。

Ascend-cann-toolkit install success.

2.3.3 设置环境变量

source /usr/local/Ascend/ascend-toolkit/set_env.sh

安装算子包后，若系统出现以下关键回显信息，表示算子包安装成功。

Ascend-cann-kernels install success.

2.3 运行Docker容器

下列命令将创建一个名为'cann-container'的Docker容器，并将设备和驱动挂载到容器中。

1. 卸载

卸载CANN-toolkit

~/Ascend/ascend-toolkit//{arch}-linux/script/uninstall.sh

卸载固件

sudo /usr/local/Ascend/firmware/script/uninstall.sh

卸载驱动

sudo /usr/local/Ascend/driver/script/uninstall.sh