AMD显卡 ROCm+ComfyUI部署全指南

目前AI画图的主流工具中,ComfyUI 绝对是佼佼者。比起我之前介绍过的 SD WebUI(详情可戳:AI绘画Stable Diffusion入门),ComfyUI 最大的优势就是支持可视化的工作流配置,灵活度拉满。

ROCm 环境下搭建 ComfyUI,实现从部署到出图的全流程。

StabilityMatrix#

对于想快速上手的同学,我推荐用 StabilityMatrix 这个工具来部署 ComfyUI,省去手动配置环境的繁琐步骤。下面就详细说下操作流程。

添加package#

打开 StabilityMatrix 后,直接在包管理界面搜索并添加 ComfyUI 包即可。对于新手来说,这个方式最省心,能快速完成基础部署。

不过这里提一句:如果是有一定技术基础的同学,我更推荐直接去 GitHub 克隆 ComfyUI 官方项目(官方地址),后续自定义配置和功能扩展会更灵活。

设置选项#

进入高级选项设置界面,核心是要选择 ROCm 环境,这样才能让 ComfyUI 正确调用 AMD 显卡的算力。(其实后面的环境还要自己卸载内置的 PyTorch 然后安装我们自己编译的 PyTorch)

这里要吐槽一下模型共享功能:这部分设置比较繁琐,而且对于大多数用户来说,后续大概率只会用 ComfyUI 这一个UI工具,基本用不上模型共享。这也是我推荐有基础的同学直接克隆 GitHub 项目的原因之一,能避开这些不必要的配置麻烦。

设置参数#

为了让 ComfyUI 在 AMD 显卡 + ROCm 环境下稳定运行,避免出现显存不足、性能拉胯等问题,我整理了一套最优启动参数,直接对照着打钩:

  • --normalvram:常规显存模式,适配多数中端AMD显卡(如RX 6650 XT)
  • --preview-method auto:自动选择预览方式,平衡预览速度和资源占用
  • --use-pytorch-cross-attention:使用PyTorch交叉注意力机制,提升生成效率
  • --fp16-vae:VAE采用FP16精度,减少显存占用的同时保证画质
  • --disable-xformers:禁用xformers(ROCm环境对xformers支持不完善,禁用后更稳定)

启动#

一切就绪,启动!

虽然报错了有点小尴尬,但是日志中的 Device: cuda:0 AMD Radeon RX 6650 XT : native 这一行非常迷人

证明之前手工编译的 PyTorch 已经完美识别并驱动了显卡,gfx1032 架构也识别正确

目前的报错 ModuleNotFoundError: No module named 'torchsde' 还是因为缺乏依赖

手动进入环境里补全一下依赖

复制代码
cd D:\Softwares\StabilityMatrix\Data\Packages\ComfyUI
.\venv\Scripts\activate
pip install -r requirements.txt

这次完美启动了!

Comfy-UI#

ComfyUI 启动成功后,在浏览器中访问日志提示的地址(默认是 http://127.0.0.1:8188),就能进入 ComfyUI 的操作界面。

好家伙,第一眼我不禁感叹,现在的工具链已经如此完善了!

可视化的工作流界面清晰明了,每个功能模块都能直观看到,操作逻辑也很顺畅。

几年前我第一次玩AI画图时,界面还是这样的:

相比之下简直降维打击啊!

从早期繁琐的参数配置界面,到如今直观的可视化工作流,能明显感受到开源社区的进步------让AI画图的门槛越来越低,同时灵活度却越来越高。

工作流#

ComfyUI 和传统的 AI 画图工具最大的区别,就是它是"工作流驱动"的------所有的生成逻辑都通过工作流来定义,你可以像搭积木一样,自由组合不同的功能模块,实现从提示词到最终图像的全流程定制。

对于新手来说,不用急着自己搭建工作流,ComfyUI 内置了丰富的模板库,直接选用现成的模板就能快速上手。我这次用的是模板库中 SD1.5 模型的「图像生成」基础模板,适合入门尝试。

加载模板后,工作流的初始形态就出来了,包含了提示词输入、模型选择、采样器配置、图像输出等核心模块,后续可以根据自己的需求调整参数或添加模块:

开始AI画图#

工作流加载完成后,不用做太多调整,直接使用模板内置的提示词就能开始生成。点击界面右上角的「运行」按钮,ComfyUI 就会调用 AMD 显卡的算力开始画图。

从任务管理器中能清晰看到,AMD 显卡的利用率直接拉满------这才是真正发挥出了 GPU 的性能优势!相比之前的 CNN 训练任务,AI 画图对算力的需求更高,也更能体现出 ROCm 环境搭建的价值。

输出效果#

使用内置模板的提示词:beautiful scenery nature glass bottle landscape, purple galaxy bottle (美丽的自然风景玻璃瓶景观,紫色银河瓶)

等待几十秒后,生成的图像效果超出预期------色彩搭配和谐,细节丰富,玻璃瓶的质感和银河的梦幻感都表现得不错,对于第一次上手的新手来说,这个结果已经很让人满意了。

小结#

本次我们成功在 Windows + AMD 显卡 + ROCm 环境下,通过 StabilityMatrix 快速部署了 ComfyUI,并完成了第一次 AI 画图。整个流程下来,最大的感受就是:如今 AMD 显卡的 AI 生态已经越来越完善,曾经"N卡专属"的 AI 画图场景,A卡用户通过 ROCm 也能轻松实现,而且体验并不逊色。

相关推荐
YJlio12 小时前
1.7 通过 Sysinternals Live 在线运行工具:不下载也能用的“云端工具箱”
c语言·网络·python·数码相机·ios·django·iphone
l1t12 小时前
在wsl的python 3.14.3容器中使用databend包
开发语言·数据库·python·databend
山塘小鱼儿13 小时前
本地Ollama+Agent+LangGraph+LangSmith运行
python·langchain·ollama·langgraph·langsimth
码说AI13 小时前
python快速绘制走势图对比曲线
开发语言·python
wait_luky14 小时前
python作业3
开发语言·python
Python大数据分析@15 小时前
tkinter可以做出多复杂的界面?
python·microsoft
大黄说说15 小时前
新手选语言不再纠结:Java、Python、Go、JavaScript 四大热门语言全景对比与学习路线建议
java·python·golang
小小张说故事15 小时前
SQLAlchemy 技术入门指南
后端·python
我是章汕呐16 小时前
拆解Libvio.link爬虫:从动态页面到反爬对抗的实战解析
爬虫·python