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