该 Skill 用于在华为昇腾 NPU 上配置 HuggingFace Diffusers 开发环境,主要面向需要运行扩散模型的昇腾用户。
用途:指导完成从 CANN 环境激活到 Diffusers 安装与验证的完整流程。
主要能力:覆盖 CANN 版本检测与激活、PyTorch + torch_npu 安装、Diffusers 库安装,以及环境验证脚本的运行;并提供 CANN 版本说明、安装细节与故障排查参考。
使用方式:先激活 CANN 环境(脚本会检测 /usr/local/Ascend/cann 与 ascend-toolkit 两种路径),再安装 PyTorch 与 torch_npu,随后安装 Diffusers 与 transformers,最后运行验证脚本 python scripts/validate_environment.py。torch_npu 可通过 pip install torch-npu 自动匹配已安装的 PyTorch 版本。
兼容性:CANN 需 8.0.RC1 及以上;PyTorch 支持 2.1.0 至 2.8.0;Diffusers 需 0.28.0 及以上,支持 SDXL、SD3、Flux。文档给出 CANN 8.3.RC1 下的配套示例,如 PyTorch 2.8.0 对应 torch_npu 2.8.0。
验证检查项包括:CANN 安装与环境变量、PyTorch 导入与版本匹配、torch_npu 导入与 NPU 可见性、Diffusers 导入、numpy 版本需低于 2.0。限制:如 torch_npu 安装失败,需查阅 torch_npu Release 的完整版本配套表;具体性能表现与其他平台支持情况暂未提供。