用途:为在昇腾NPU平台上安装CANN(神经网络计算架构)提供分步指导,适用于用户询问CANN安装、配置昇腾开发环境或排查安装问题。
主要能力与流程:
- 驱动检查:使用 npu-smi info 确认NPU设备与驱动状态,并记录驱动版本用于CANN版本兼容性检查;若提示 command not found 需先安装昇腾驱动。
- 依赖检查:确认 python3/python 与 pip3/pip 是否可用,缺失时参考环境依赖安装指南。
- 安装CANN:离线方式需当前目录存在 Ascend-cann-toolkit_.run 与 Ascend-cann--ops_*.run,先装Toolkit再装ops,命令为 bash "$TOOLKIT_PACKAGE" --install --quiet 与 bash "$OPS_PACKAGE" --install --quiet,默认路径 /usr/local/Ascend;conda方式先创建虚拟环境(支持Python 3.7.x-3.13.11),添加昇腾源 https://repo.huaweicloud.com/ascend/repos/conda,再 conda install ascend::cann-toolkit 与 conda install ascend::cann-{芯片类型}-ops;yum方式配置华为源后安装 Ascend-cann-toolkit 与 Ascend-cann-{芯片类型}-ops。
- 环境变量:临时 source /usr/local/Ascend/ascend-toolkit/set_env.sh,永久写入 ~/.bashrc。
- 验证:python3 -c "import acl;acl.init();acl.rt.set_device(0)"。
- 运行时依赖:安装 attrs、cython、numpy、decorator、sympy、cffi、pyyaml、pathlib2、psutil、protobuf==3.20.0、scipy、requests、absl-py 等。
限制与注意:需已安装且状态正常的昇腾NPU设备;CANN版本须与驱动版本匹配;必须先装Toolkit再装ops;安装后必须设置环境变量;部分操作可能需要root权限。故障排查建议检查驱动兼容性、依赖完整性与安装日志,或参考昇腾社区文档。具体芯片类型、版本号与详细依赖安装命令暂未提供。