在Linux上运行ComfyUI比在Windows上运行更快更好,接下来将演示如何在AMD显卡(RX 6600 XT)下进行安装。
Python版本
Github: https://github.com/comfyanonymous/ComfyUI
ComfyUI推荐的版本是3.12,如果不想降级的话,使用pyenv来进行版本控制。
1 2 3 4 5 6 7 8 9 10 11 12
| sudo pacman -S pyenv
pyenv install 3.12.0
git clone https://github.com/pyenv/pyenv-virtualenv.git $(pyenv root)/plugins/pyenv-virtualenv
pyenv virtualenv 3.12 venv
source /home/username/.pyenv/versions/venv/bin/activate
deactivate
|
安装与启动
根据官方文档的步骤进行安装即可,一行代码就能安装完成。
1 2 3 4
| pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.4
HSA_OVERRIDE_GFX_VERSION=10.3.0 python main.py
|
总结
安装并不难,使用官方推荐的Python版本基本不会报错。请注意不同显卡的安装命令不同,详情请看官方文档。