Diffusers 昇腾 NPU 环境配置

本 Skill 指导用户在华为昇腾 NPU 上配置 HuggingFace Diffusers 开发环境。

快速开始

4 步完成环境搭建：

# 1. 激活 CANN 环境（自动检测版本）
if [ -d "/usr/local/Ascend/cann" ]; then
    source /usr/local/Ascend/cann/set_env.sh
else
    source /usr/local/Ascend/ascend-toolkit/set_env.sh
fi

# 2. 安装 PyTorch + torch_npu（版本自动匹配）
pip install torch --index-url https://download.pytorch.org/whl/cpu
pip install torch-npu  # 自动匹配已安装的 PyTorch 版本

# 3. 安装 Diffusers
pip install diffusers["torch"] transformers

# 4. 验证环境
python scripts/validate_environment.py

安装概览

步骤	详细指南
CANN 验证	references/cann-versions.md
PyTorch + torch_npu	references/installation.md
Diffusers 安装	references/installation.md

版本兼容性

组件	版本	说明
CANN	8.0.RC1+	NPU 支持必需
PyTorch	2.1.0 - 2.8.0	含 torch_npu 扩展
Diffusers	0.28.0+	支持 SDXL、SD3、Flux

torch_npu 版本匹配：

• 默认安装与 PyTorch 版本相同的 torch_npu：pip install torch-npu
• 如安装失败，请参考 torch_npu Release 查看完整版本配套表

示例（CANN 8.3.RC1）：

PyTorch	torch_npu	Python
2.8.0	2.8.0	3.9 - 3.11
2.7.1	2.7.1	3.9 - 3.11
2.6.0	2.6.0.post3	3.9 - 3.11

环境验证

运行验证脚本：

python scripts/validate_environment.py

检查项：

检查项	说明
CANN 安装	目录存在且环境变量已设置
PyTorch	import 成功，版本匹配
torch_npu	import 成功，NPU 可见
Diffusers	import 成功
numpy	版本 < 2.0

参考资源

• 详细安装指南 - 完整的安装步骤和命令
• CANN 版本说明 - 版本检测和差异说明
• 故障排查 - 常见问题和解决方案
• Diffusers 官方文档
• 昇腾 PyTorch 扩展
• torch_npu Release