用途:该技能面向昇腾NPU环境配置,提供从驱动安装到Python环境配置的操作流程,用于在安装昇腾驱动、CANN工具包、torch_npu或处理相关环境问题时参考执行。
主要能力:
- 版本配套查询:给出CANN、PyTorch、torch_npu、Python以及CANN与Ascend HDK(驱动固件)之间的版本对应关系快照,并说明按PyTorch版本、torch_npu版本、CANN版本、驱动版本的顺序确定配套版本的选择原则,附有多个版本选择示例。
- Docker镜像方式搭建(推荐):从昇腾镜像仓库选择CANN基础镜像,说明镜像Tag命名规则(CANN版本-芯片型号-OS-pyPython版本)与拉取命令,并给出启动容器时挂载NPU设备的完整docker run命令及参数说明。
- 容器内配置:配置CANN与NNAL环境变量,安装PyTorch与torch_npu,并用torch.npu.is_available()验证。
- 手动安装方式(备选):依次覆盖昇腾驱动与固件安装、CANN toolkit与kernels安装、环境变量配置、conda或pip创建Python环境、PyTorch与torch_npu安装及验证代码。
- 环境变量配置:列出必要的CANN环境变量、日志级别、可见设备设置以及算子编译、内存和HCCL通信等性能优化变量。
- 第三方库安装:说明优先查阅官方第三方库支持列表,列出transformers、accelerate、peft、trl、deepspeed等的适配情况,并提供FFmpeg、OpenCV及通用库的安装建议与编译模板。
- 资源获取:建议通过ModelScope或HuggingFace镜像站获取模型与数据集,避免默认依赖HuggingFace官网。
使用方式:以技能形式在安装昇腾驱动、CANN、torch_npu或排查环境问题时调用;实际执行时须先访问昇腾官网获取最新版本配套关系,仅在无法访问官网时回退使用技能内的快照数据。环境搭建优先采用Docker镜像方式,无法使用Docker时再按手动流程安装。
适用场景:昇腾NPU服务器或开发机的环境初始化、驱动与CANN安装、torch_npu配置、容器化开发环境搭建、ARM(aarch64)架构下第三方库安装以及内网环境下的模型资源获取。
限制:版本配套信息为编写时快照,需以官网最新文档为准;手动安装需root权限并重启系统;宿主机需已安装昇腾驱动和固件且Docker版本不低于1.11.2;部分第三方库在ARM架构下可能无预编译包;技能内的具体镜像Tag和下载链接随版本更新,需查阅官方渠道确认。