开发者
资源

安装NPU驱动(Yum在线安装)

本节将介绍通过Yum快速安装NPU驱动的操作步骤。

准备用户

本节指导用户创建安装和运行用户,若用户已存在,请跳过本节内容。

表1 用户类型

组件

安装用户

运行用户

NPU驱动

root

  • 由于安装驱动时,运行用户和用户组默认指定为HwHiAiUser,需在安装软件包前自行创建HwHiAiUser的运行用户和用户组。
  • 若创建的用户和用户组是非HwHiAiUser(含root),安装驱动时必须指定运行用户。

本文档中的示例步骤使用的用户如下:

  • 以root用户安装驱动。
  • 以非root用户HwHiAiUser(驱动的默认运行用户)为运行用户。
执行如下示例命令,创建HwHiAiUser用户和用户组:
1
2
groupadd HwHiAiUser
useradd -g HwHiAiUser -d /home/HwHiAiUser -m HwHiAiUser -s /bin/bash

默认用户HwHiAiUser的用途和配置要求,请以Atlas 系列硬件产品 账户清单为准,若想创建其他非root用户,请用户自行创建。

安装依赖和配置昇腾源

  1. 执行如下命令检查源是否可用。
    1
    yum makecache
    

    如果命令执行报错或者等待时间过长,则检查网络是否连接或修改“/etc/yum.repos.d/xxxx.repo”文件为可用源(以配置华为镜像源为例,可参考华为开源镜像站中镜像源对应的配置方法操作)。

  2. 执行命令安装所需依赖
    1
    yum install -y make dkms gcc kernel-headers-$(uname -r) kernel-devel-$(uname -r)
    

    如果出现报错或者依赖不存在,请参见安装驱动源码编译所需依赖解决。

  3. 配置昇腾源,否则将无法安装。
    curl https://repo.oepkgs.net/ascend/cann/ascend.repo -o /etc/yum.repos.d/ascend.repo && yum makecache

安装NPU驱动

表2 安装驱动

产品类型

安装命令

Atlas A3 系列产品

1
yum install -y Atlas-A3-hdk-npu-driver-25.5.0

Atlas A2 系列产品

1
yum install -y Ascend910B-driver-25.5.0

Atlas 训练系列产品

1
yum install -y Ascend-hdk-910-npu-driver-25.5.0

Atlas 推理系列产品

1
yum install -y Ascend-hdk-310p-npu-driver-25.5.0
安装完成后,可根据系统提示信息决定是否重启系统,若需要重启,请执行reboot命令,并执行如下命令查看驱动加载是否成功。
1
npu-smi info

若返回驱动相关信息说明加载成功。否则,说明加载失败,请访问支持与服务在论坛求助或提交技术工单。