昇腾-comfyui适配&推理
·
省时方案:基于mindie环境搭建comfyui环境,因为mindie环境各个组件配套;
1、获取mindie镜像:
docker pull swr.cn-south-1.myhuaweicloud.com/ascendhub/mindie:2.3.0-800I-A2-py311-openeuler24.03-lts
2、拉起容器(需要暴露端口便于访问):
docker run -it -d --shm-size=200g \
--privileged \
--name comfyui \
--device=/dev/davinci_manager \
--device=/dev/hisi_hdc \
--device=/dev/devmm_svm \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver:ro \
-v /usr/local/sbin:/usr/local/sbin:ro \
-v /data:/data \
-p 8188:8188 \
swr.cn-south-1.myhuaweicloud.com/ascendhub/mindie:2.3.0-800I-A2-py311-openeuler24.03-lts \
/bin/bash
进入容器
docker exec -it comfyui bash
3、安装miniconda
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-latest-Linux-aarch64.sh
bash Miniconda3-latest-Linux-aarch64.sh
安装过程中全选yes
source ~/.bashrc
4、安装依赖
yum update
yum install -y gcc gcc-c++ make cmake zlib zlib-devel openssl openssl-devel sqlite-devel libffi-devel bzip2-devel libxslt-devel unzip pciutils net-tools blas-devel gcc-gfortran vim zip wget git
error1
[root@4e676ad44ce5 miniconda]# source ~/.bashrc
Error while loading conda entry point: conda-anaconda-tos (No module named 'pydantic_core._pydantic_core')
Error while loading conda entry point: anaconda-auth (No module named 'pydantic_core._pydantic_core')
(Miniconda 最新版本中集成的 conda-anaconda-tos 和 anaconda-auth 插件与当前环境(尤其是 ARM64 架构)存在兼容性问题,不影响主要功能使用)
5、python环境创建
conda create -n comfyui python=3.11
conda activate comfyui
echo "conda activate comfyui" >> ~/.bashrc
6、安装依赖
pip3 install attrs numpy==1.26.4 decorator sympy cffi pyyaml pathlib2 psutil protobuf scipy requests absl-py -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install torchvision==0.20.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install torch==2.5.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install torch_npu==2.5.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip3 install torch_npu==2.5.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install color-matcher
pip install imageio-ffmpeg
yum install -y ffmpeg
检查
cd (comfyui) [root@4e676ad44ce5 /]# pip list | grep torch
torch 2.5.1
torch_atb 0.0.1
torch-npu 2.5.1
torchaudio 2.5.1
torchsde 0.2.6
torchvision 0.20.1
7、安装comfyui
git clone https://gitee.com/mirrors/comfyui.git
cd comfyui
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
安装插件
cd custom_nodes
git clone https://gitee.com/zuohaozhan/ComfyUI-Manager.git
git clone https://gitee.com/comfyuiwiki/ComfyUI-VideoHelperSuite.git
git clone https://gitee.com/xeval/ComfyUI-KJNodes.git
8、修改代码
vim main.py
# 添加以下内容
import torch
import torch_npu
torch.npu.config.allow_internal_format = False
torch.npu.set_compile_mode(jit_compile=False)
error2
Error while loading conda entry point: conda-anaconda-tos (No module named 'pydantic_core._pydantic_core')
Error while loading conda entry point: anaconda-auth (No module named 'pydantic_core._pydantic_core')
cd Error while loading conda entry point: conda-anaconda-tos (No module named 'pydantic_core._pydantic_core')
Error while loading conda entry point: anaconda-auth (No module named 'pydantic_core._pydantic_core')
解决:
pip install pydantic pydantic-core --force-reinstall
9、拉起comfyui服务
python main.py --listen 0.0.0.0 --port 8188
10、访问ui界面
http://localhost:8188/

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐

所有评论(0)