這邊記錄我在 Linux 藉由 AMD ROCm 成功部署 ComfyUI 啟用 Krita AI Diffusion,未來重新部署也能使用,希望對有 AMD 卡但是可能對於指令操作不這麼熟悉的用戶有點幫助。
不想搞這麼麻煩,最容易省時的方法是,直接買張 Nvidia 的顯示卡,記憶體最好16G以上 ,裝好 Krita 外掛就直接跑自動安裝,但是不開光追下,以娛樂及跑圖綜合用途來看的情況下 Nvidia 顯示卡價錢要多快一倍。
我使用的是 Kbuntu(Ubuntu) 24.04
顯示卡是 RX7900GRE 16G
步驟一:安裝 AMD 驅動及 ROCm
BIOS 設定
安裝前,重新開機一直按del進入bios,把主機板Bios裡的安全啓動關閉,後續安裝套件比較方便。(Secure Boot:每家主機板廠商位置不太一樣,關鍵字就是Secure Boot/安全開機)
安裝 ROCm 驅動
完成後重開機,終端機輸入以下確認 ROCm 是否正確啟用:
安裝完後,關閉終端機並輸入以下,指令禁止 conda 自動啟動 base 環境:
建立名為“ai”的虛擬環境(Python 3.10):

步驟三:安裝 PyTorch(支援 ROCm)
PyTorch網站頁面有自動生成的指令:https://pytorch.org/get-started/locally/
安裝對應 ROCm 6.3 的 PyTorch:
驗證安裝是否成功:

步驟四:部署 ComfyUI
這邊可以找到自己顯示卡版本:https://rocm.docs.amd.com/en/latest/reference/gpu-arch-specs.html
確認顯示卡版本後,替換下面的版本數字(例如 gfx1100對應11.0.0)輸入會啓用ComfyUI :
ComfyUI 預設開啟網址為:http://127.0.0.1:8188/
重新啟動 ComfyUI:

參考:https://github.com/Acly/krita-ai-diffusion/wiki/ComfyUI-Setup
使用 ComfyUI Manager 安裝相依的 Custom Nodes 與模型。

步驟五:安裝 Krita 及 Krita AI Diffusion 外掛

每次使用需啟動 ComfyUI

不想搞這麼麻煩,最容易省時的方法是,直接買張 Nvidia 的顯示卡,記憶體最好16G以上 ,裝好 Krita 外掛就直接跑自動安裝,但是不開光追下,以娛樂及跑圖綜合用途來看的情況下 Nvidia 顯示卡價錢要多快一倍。
我使用的是 Kbuntu(Ubuntu) 24.04
顯示卡是 RX7900GRE 16G
步驟一:安裝 AMD 驅動及 ROCm
在終端機使用以下指令,更新linux套件
sudo apt update |
BIOS 設定
安裝前,重新開機一直按del進入bios,把主機板Bios裡的安全啓動關閉,後續安裝套件比較方便。(Secure Boot:每家主機板廠商位置不太一樣,關鍵字就是Secure Boot/安全開機)
安裝 ROCm 驅動
參考官方網站安裝指令:
安裝ROCm
wget https://repo.radeon.com/amdgpu-install/6.4/ubuntu/noble/amdgpu-install_6.4.60400-1_all.deb sudo apt install ./amdgpu-install_6.4.60400-1_all.deb sudo apt update sudo apt install python3-setuptools python3-wheel sudo usermod -a -G render,video $LOGNAME # Add the current user to the render and video groups sudo apt install rocm |
安裝驅動
wget https://repo.radeon.com/amdgpu-install/6.4/ubuntu/noble/amdgpu-install_6.4.60400-1_all.deb sudo apt install ./amdgpu-install_6.4.60400-1_all.deb sudo apt update sudo apt install "linux-headers-$(uname -r)" "linux-modules-extra-$(uname -r)" sudo apt install amdgpu-dkms |
完成後重開機,終端機輸入以下確認 ROCm 是否正確啟用:
rocminfo |
步驟二:安裝 Miniconda 後建立虛擬 Python 開發環境
在終端機使用以下指令,安装miniconda。(安裝期間會需要在終端機輸入yes、按enter下拉,再輸入yes。)
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh bash ./Miniconda3-latest-Linux-x86_64.sh |
安裝完後,關閉終端機並輸入以下,指令禁止 conda 自動啟動 base 環境:
conda config --set auto_activate_base false |
建立名為“ai”的虛擬環境(Python 3.10):
conda create -n ai python=3.10 -y conda activate ai |
開啓後終端機畫面會如以下
步驟三:安裝 PyTorch(支援 ROCm)
PyTorch網站頁面有自動生成的指令:https://pytorch.org/get-started/locally/
我在網站選擇的是:
- stable (版本會更新,我選stable版本)
- Linux
- Pip
- Python
- ROCm 6.3
這邊有時候會下載失敗,就要重新跑,整個流程時間比較久。
conda activate ai pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.3 |
驗證安裝是否成功:
conda activate ai python3 -c 'import torch' 2> /dev/null && echo 'Success' || echo 'Failure' python3 -c 'import torch; print(torch.cuda.is_available())' |
步驟四:部署 ComfyUI
在終端機使用以下指令,部署ComfyUI
conda activate ai git clone https://github.com/comfyanonymous/ComfyUI.git cd ~/ComfyUI pip install -r requirements.txt |
這邊可以找到自己顯示卡版本:https://rocm.docs.amd.com/en/latest/reference/gpu-arch-specs.html
確認顯示卡版本後,替換下面的版本數字(例如 gfx1100對應11.0.0)輸入會啓用ComfyUI :
HSA_OVERRIDE_GFX_VERSION=11.0.0 python3 main.py |
ComfyUI 預設開啟網址為:http://127.0.0.1:8188/
安裝 ComfyUI Manager
安裝ComfyUI Manager
cd ~/ComfyUI/custom_nodes conda activate ai git clone https://github.com/ltdrdata/ComfyUI-Manager comfyui-manager |
重新啟動 ComfyUI:
conda activate ai cd ~/ComfyUI HSA_OVERRIDE_GFX_VERSION=11.0.0 python3 main.py |
使用 ComfyUI Manager 安裝相依的 Custom Nodes 與模型。
可從 Windows使用外掛自動下載後搬移到此處,自己額外下載的模型可以丟到checkpoints。
如果是N卡的使用者,可以直接在windows下從這邊開始
- 安裝 Krita:https://krita.org/zh-tw/download/
- 外掛下載與安裝:https://github.com/Acly/krita-ai-diffusion/releases/tag/v1.31.2
- Krita 版本需 >= 5.2
- 外掛設定中使用預設伺服器網址即可連線。
每次開機或者關閉終端機,都要自己手動輸入以下啓用comfyui,Krita ai才能連上伺服器,也可以做腳本跑,之後就可以快樂的出圖了。
cd ~/ComfyUI conda activate ai HSA_OVERRIDE_GFX_VERSION=11.0.0 python3 main.py |