昇腾AI服务器基础配置:驱动、固件与NPU识别检查
在昇腾AI服务器上开始跑模型之前,核心配置任务是把NPU驱动和CANN工具包装好,并确认系统能正确识别加速卡。
本文按实际运维顺序,从环境核对到设备验证逐步展开,适合第一次接触昇腾服务器的读者。
先核对系统与硬件兼容性
动手安装前,先确认服务器型号、NPU卡型号和操作系统版本在厂商兼容列表内。
昇腾驱动和固件对内核版本有要求,混用不匹配的版本可能导致设备无法识别。
登录服务器后执行以下命令收集基础信息:
uname -r
cat /etc/os-release
lspci | grep -i processing
lspci 能列出加速卡是否被PCIe总线发现。
如果这里看不到设备,先检查卡是否插紧、BIOS是否启用对应插槽,不要急着装驱动。
另外记录当前固件版本,后续升级时用得上。
安装NPU驱动和固件
从昇腾社区或服务器厂商支持页面下载与卡型号、系统版本匹配的驱动包和固件包。
安装前建议停掉占用设备的业务进程,并备份当前驱动版本信息。
以常见的.run安装包为例,先给执行权限再运行:
chmod +x Ascend-hdk-*-npu-driver_*.run
./Ascend-hdk-*-npu-driver_*.run --full
安装过程中会提示是否重启,按提示操作。
固件包通常也以.run形式提供,执行类似命令即可。
安装完成后用 npu-smi info 查看设备列表,正常应显示卡编号、型号和健康状态。
若提示命令不存在,检查环境变量或安装路径是否已加入PATH。
安装CANN工具包并验证
驱动就绪后安装CANN,它提供算子库、编译工具和运行时。
下载对应版本的CANN包,解压后运行安装脚本:
./Ascend-cann-toolkit_*.run --install
安装完成后设置环境变量,通常需要source安装目录下的set_env.sh:
source /usr/local/Ascend/ascend-toolkit/set_env.sh
验证CANN是否可用,可以运行自带的环境检查工具:
ascend-dmi -i
该命令会输出设备健康、驱动版本和CANN版本信息。
如果显示设备状态异常,优先回看驱动与固件版本是否一致。
常见问题与避坑提醒
- npu-smi无输出或报错:先确认驱动安装成功,再检查
/dev/davinci*设备节点是否存在。 - 版本不匹配:驱动、固件、CANN三者版本需按官方配套关系选择,不要只升级其中一个。
- 权限问题:普通用户执行npu-smi可能权限不足,按厂商文档将用户加入对应用户组。
- 内核升级后失效:系统升级内核后需重新编译驱动模块,建议升级前确认驱动支持新内核。
验证配置结果
完成以上步骤后,用 npu-smi info 确认所有卡均被识别且状态正常,再用 ascend-dmi -i 检查整体环境。
若计划运行AI框架,可进一步安装对应版本的PyTorch或MindSpore适配包,并跑一个简单算子测试确认NPU可被框架调用。
配置完成后建议记录驱动、固件和CANN版本,便于后续排错和升级对照。