ComfyUI在昇腾上的实践
本文介绍了在昇腾AI硬件平台上部署ComfyUI服务的详细步骤。首先列出硬件和软件环境要求，包括Atlas 800T/I A2设备、64GB NPU显存及相应驱动版本。安装过程包含克隆项目、修改依赖文件、安装Python包等步骤，特别指出了升腾平台不支持fp8_e4m3fn类型模型的注意事项。最后提供了模型下载和工作流验证方法，指导用户将下载的模型文件放置到正确目录并运行验证。整个过程详细说明了在
一、安装
先置条件：
- Atlas 800T/I A2
- npu显存： 64GB
- 驱动： HDK 25.3.rc1
- CANN: 8.3.RC1
- 推荐docker镜像：ascend/cann:8.3.rc1-910b-openeuler24.03-py3.11
- pytorch: 2.7.1
- PTA: 2.7.1
注意事项：昇腾加载fp8_e4m3fn类型模型，会报错，暂不支持该类型模型，使用工作流模板时，替换为fp16类型模型即可
参考文档：ComfyUI链接
操作步骤：
- 参考文档下载项目
git clone https://github.com/comfyanonymous/ComfyUI.git
- 修改requirements.txt依赖文件
直接安装，transformers会搜索并安装torch的最新版本，覆盖容器内的原版本，所以可以把transformers的依赖提前，后续指定torch的版本安装，覆盖最新版本
comfyui-frontend-package==1.28.8
comfyui-workflow-templates==0.2.11
comfyui-embedded-docs==0.3.1
numpy>=1.25.0
einops
transformers>=4.37.2
tokenizers>=0.13.3
sentencepiece
safetensors>=0.4.2
aiohttp>=3.11.8
yarl>=1.18.0
pyyaml
Pillow
scipy
tqdm
psutil
alembic
SQLAlchemy
av>=14.2.0
torch==2.7.1
torchsde
torchvision==0.22.1
torchaudio==2.7.1
#non essential dependencies:
kornia>=0.7.1
spandrel
pydantic~=2.0
pydantic-settings~=2.0
- 安装依赖
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
- 部署ComfuUI服务
python main.py --listen <ip 地址>
二、模型生成验证
参考WAN2.2 14B Text to Video工作流模板
模型下载：modelscope下载页面
模型下载以后，根据目录对应分发到ComfyUI/models下的对应目录下
audio_encoders -> models/audio_encoders
diffusion_models -> models/diffusion_models
loras -> models/loras
text_encoders -> models/text_encoders
vae -> models/vae
点击运行，视频生成验证
鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者，聚合华为+生态”的社区，内容涵盖鲲鹏、昇腾资源，帮助开发者快速获取所需的知识、经验、软件、工具、算力，支撑开发者易学、好用、成功，成为核心开发者。
更多推荐
5
0- 0
已为社区贡献2条内容
所有评论(0)