快速安装昇腾环境

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

1. 系统要求

1.1 前置检查

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

lspci | grep 'Processing accelerators'

确认操作系统架构及版本

uname -m && cat /etc/*release

确认Python版本

python --version

1.2 软件要求

软件	版本
操作系统	openEuler20.03/22.03, Ubuntu 20.04/22.04
Python	3.8, 3.9, 3.10

2. 环境安装

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

警告

以下文档需要使用非root用户进行安装安装

安装方式

操作系统

操作系统版本

CPU架构

NPU型号

昇腾套件版本

安装方式

直接安装

Docker

操作系统

openEuler

Ubuntu

操作系统版本

CPU架构

x86-64

aarch64

NPU型号

Atlas 800T A2 训练卡

Atlas 300I Pro 推理卡

昇腾套件版本

Select CANN Version

Driver

Firmware

2.1 安装驱动

2.1.1 安装依赖

sudo apt-get install -y gcc g++ make cmake zlib1g zlib1g-dev openssl libsqlite3-dev libssl-dev libffi-dev unzip pciutils net-tools libblas-dev gfortran libblas3 python3-dev

sudo yum install -y gcc gcc-c++ make cmake unzip zlib-devel libffi-devel openssl-devel pciutils net-tools sqlite-devel lapack-devel gcc-gfortran python3-devel

2.1.2 创建驱动运行用户

备注

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

sudo groupadd -g 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 numpy decorator sympy cffi pyyaml pathlib2 psutil protobuf scipy requests absl-py wheel typing_extensions

2.3.2 下载并安装

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

Ascend-cann-toolkit install success.

2.3.3 安装算子包

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

Ascend-cann-kernels install success.

2.3.4 设置环境变量

echo "source ~/Ascend/ascend-toolkit/set_env.sh" >> ~/.bashrc
source ~/.bashrc

2.3 运行Docker容器

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

3. 卸载

卸载驱动

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

卸载固件

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

卸载CANN-toolkit

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