快轉到主要內容

🖥️ 安裝至GNU/Linux|Stable Diffusion WebUI使用手冊

分類   開源AI應用 AI繪圖教學
🗓️ 民國113年 甲辰年
✍ 切換正體/簡體字
目錄
Stable Diffusion WebUI使用手冊 - 點選展開系列文章
5: 本篇文章

本文以Ubuntu 24.04 LTS為例子。

請使用一般使用者,不可以用Root使用者安裝。

此圖使用Stable Diffusion WebUI生成

1 安裝顯示卡驅動
#

1.1. Nvidia與CUDA
#

  1. 安裝 專有Nvidia驅動,之後重開機。

  2. 安裝 CUDA

  3. 確認驅動安裝是否正確安裝

nvidia-smi
nvcc --version

1.2. AMD與AMD ROCm
#

  1. 安裝AMD驅動程式,重開機。
sudo apt update && sudo apt upgrade -y
wget https://repo.radeon.com/amdgpu-install/22.40.3/ubuntu/focal/amdgpu-install_5.4.50403-1_all.deb
sudo apt-get install ./amdgpu-install_5.4.50403-1_all.deb
  1. 安裝AMD ROCm,中間需要重開機。
sudo apt install linux-headers-`uname -r` linux-modules-extra-`uname -r`
curl -fsSL https://repo.radeon.com/rocm/rocm.gpg.key|sudo gpg --dearmor -o /etc/apt/trusted.gpg.d/rocm-keyring.gpg
echo 'deb [arch=amd64 signed-by=/etc/apt/trusted.gpg.d/rocm-keyring.gpg] https://repo.radeon.com/amdgpu/5.4.2/ubuntu jammy main'|sudo tee /etc/apt/sources.list.d/amdgpu.list
sudo apt-get update
sudo apt install amdgpu-dkms
sudo reboot
echo 'deb [arch=amd64 signed-by=/etc/apt/trusted.gpg.d/rocm-keyring.gpg] https://repo.radeon.com/rocm/apt/5.4.2 jammy main'|sudo tee /etc/apt/sources.list.d/rocm.list
echo -e 'Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600'|sudo tee /etc/apt/preferences.d/rocm-pin-600
sudo apt-get update
sudo apt install rocm-hip-sdk rocm-opencl-sdk
echo "export LD_LIBRARY_PATH=/opt/rocm-5.4.2/lib:/opt/rocm-5.4.2/lib64" >> ~/.bashrc
source ~/.bashrc
  1. 確認驅動是否有正確安裝
# 應會看到driver=amdgpu
sudo lshw -c video
# 查看ROCm是否正確安裝
sudo apt show rocm-libs

2. 安裝Anaconda
#

  1. 安裝Python、wget、git。
sudo apt install python3 python3-pip python3-virtualenv wget git
  1. 安裝 Anaconda。裝Anconda的用意是鎖Python版本,讓Stable Diffusion WebUI的Pyhon版本跟系統的隔離。

  2. 建立Python 3.10.6的虛擬環境

conda create --name sdwebui python=3.10.6

3. 複製Stable Diffusion WebUI儲存庫
#

  1. 將Stable Diffusion WebUI複製至使用者家目錄
cd ~
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
  1. (選擇性) 預先將模型下載至stable-diffusion-webui/models/Stable-diffusion目錄,這裡使用Anything v4.5
wget -nc -P ~/stable-diffusion-webui/models/Stable-diffusion https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.5-pruned.safetensors -O anything-v4.5-pruned.safetensors
  1. 啟動虛擬環境:
conda activate sdwebui

如果想要於bash script自動以conda啟動虛擬環境,可以把這二行加入至webui-user.sh頂端:

eval "$(conda shell.bash hook)"
conda activate sdwebui

於conda環境下,最好使用conda指令來安裝套件。不過Stable Diffusion WebUI仍偏好以pip建立虛擬環境。

4. 設定啟動引數
#

用VIM編輯webui-user.sh

顯示卡VRAM在8GB以上,將# COMMANDLINE_ARGS=替換成COMMANDLINE_ARGS=--xformers --enable-insecure-extension-access引數。

若顯示卡VRAM小於等於4GB,將# COMMANDLINE_ARGS=替換成COMMANDLINE_ARGS=--medvram --opt-split-attention --xformers --enable-insecure-extension-access引數。

額外加上--listen參數讓其他電腦能以IP存取網頁界面,--share則是產生一組Gradio網址。

加上--xformers可進一步減少VRAM佔用,只支援Nvidia顯示卡。加入--no-half-vae減少使用VAE時算出黑圖的機率。

加上--enable-insecure-extension-access讓你在網頁界面直接裝擴充功能。加上--nowebui則不會啟動網頁界面,單純接收WebUI API之用。

其餘可用引數請見 命令列引數

5. 啟動Stable Diffusion WebUI
#

執行webui.sh,它會自動下載PyTorch並安裝依賴套件。跑完它應該會顯示一組網址,本機用瀏覽器開啟http://127.0.0.1:7860進入網頁圖形界面。

cd ~/stable-diffusion-webui
./webui.sh

Stable Diffusion WebUI使用手冊 - 點選展開系列文章
5: 本篇文章

相關文章

命令列引數|Stable Diffusion WebUI使用手冊
分類   開源AI應用 AI繪圖教學
如何下載模型|Stable Diffusion WebUI使用手冊
分類   開源AI應用 AI繪圖教學
系統需求|Stable Diffusion WebUI使用手冊
分類   開源AI應用 AI繪圖教學

留言板

此處提供二種留言板。點選按鈕,選擇您覺得方便的留言板。要討論程式碼請用Giscus,匿名討論請用Disqus。

這是Giscus留言板,需要Github帳號才能留言。支援markdown語法,若要上傳圖片請貼Imgur或Postimages。您的留言會在Github Discussions向所有人公開。

這是Disqus留言板,您可能會看到Disqus強制投放的廣告。為防止垃圾內容,有時留言可能會被系統判定需審核,導致延遲顯示,請見諒。若要上傳圖片請善用圖床網站。