前往
大廳
主題

AMD Yes!沒N卡也能跑AI繪圖-Linux系統使用AMD卡跑Krita AI Diffusion

Lin | 2025-05-15 06:26:12 | 巴幣 12 | 人氣 167

這邊記錄我在 Linux 藉由 AMD ROCm 成功部署 ComfyUI 啟用 Krita AI Diffusion,未來重新部署也能使用,希望對有 AMD 卡但是可能對於指令操作不這麼熟悉的用戶有點幫助。

不想搞這麼麻煩,最容易省時的方法是,直接買張 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
這邊有時候會下載失敗,就要重新跑,整個流程時間比較久。

安裝對應 ROCm 6.3 的 PyTorch:
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

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

可從 Windows使用外掛自動下載後搬移到此處,自己額外下載的模型可以丟到checkpoints。


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

如果是N卡的使用者,可以直接在windows下從這邊開始
  1. 安裝 Krita:https://krita.org/zh-tw/download/
  2. 外掛下載與安裝:https://github.com/Acly/krita-ai-diffusion/releases/tag/v1.31.2
  3. Krita 版本需 >= 5.2
  4. 外掛設定中使用預設伺服器網址即可連線。


每次使用需啟動 ComfyUI

每次開機或者關閉終端機,都要自己手動輸入以下啓用comfyui,Krita ai才能連上伺服器,也可以做腳本跑,之後就可以快樂的出圖了。
cd ~/ComfyUI
conda activate ai
HSA_OVERRIDE_GFX_VERSION=11.0.0 python3 main.py



更多創作