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 HwHiAiUsersudo useradd -g HwHiAiUser -d /home/HwHiAiUser -m HwHiAiUser -s /bin/bashsudo 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
2.3.4 安装算子包
安装算子包后,若系统出现以下关键回显信息,表示算子包安装成功。
Ascend-cann-kernels install success.
2.3 运行Docker容器
下列命令将创建一个名为'cann-container'的Docker容器,并将设备和驱动挂载到容器中。