admin 管理员组

文章数量: 1184232

前置准备

确保系统满足以下条件:

  • 操作系统:Windows 10/11(64位)
  • 硬件:幻x2025 AI max+ 395(gfx1151架构)
  • 显卡驱动:25.20.01.14 graphics driver或更高版本(测试使用ROCm6.4.4推荐驱动亦可)
  • Python环境:3.12及以上(推荐使用Miniconda/Anaconda管理)
  • 工具:Git、PIP最新版

安装ROCm7与PyTorch

从AMD预览版仓库安装适配gfx1151的PyTorch套件(不知道和大神们的预编译版本有什么区别):

python -m pip install --index-url https://rocm.nightlies.amd/v2/gfx1151/ --pre torch torchaudio torchvision

我用命令直接装了Rocm7.9. 其它显卡或者芯片可以将gfx1151版本改成gfx110X-dgpu、gfx120X-all、gfx94X-dcgpu、gfx950-dcgpu

验证安装:

python -c "import torch; print(torch.cuda.is_available())"
python -c "import torch; print(f'device name [0]:', torch.cuda.get_device_name(0))"
python -m torch.utils.collect_env

配置ComfyUI环境

克隆仓库并修改依赖:

git clone https://github/comfyanonymous/ComfyUI.git
cd ComfyUI

编辑requirements.txt

  • 注释掉torchtorchaudiotorchvisionnumpy
  • 如需有问题,可指定numpy版本为numpy==1.26.4
  • 如果后面模型使用有问题也可更改transformers的版本

安装剩余依赖:

pip install -r requirements.txt

启动与使用

运行主程序:

python main.py

访问http://127.0.0.1:8188,下载所需模型后即可开始AI绘图。若遇到依赖冲突,可尝试创建干净的conda环境重新安装。

注意事项

  • 不同GPU架构需替换URL中的gfx1151为对应型号
  • 预览版可能存在稳定性问题,建议定期检查更新
  • 显存不足时可调整ComfyUI工作流中的批处理大小或更改显存设置,最高可设置96GB

参考文章
 

PyTorch via PIP installation — Use ROCm on Radeon and Ryzen
Windows部署ROCm7 rc来使用ComfyUI演示_哔哩哔哩_bilibili

本文标签: 环境 max AI AMD Windows