Stable Diffusion Web UIをローカル環境にインストール (Ubuntu)

Ubuntu20.04にStable Diffusionの実行環境を構築。

PC環境

• OS: Ubuntu 20.04 LTS

• GPU: NVIDIA GeForce RTX2070 SUPER

• NVIDIAドライバ: 535.54.03

• CUDAバージョン: 12.2

• cuDNNバージョン: 8.9.2.26

• Python: 3.8.10

インストール

依存ライブラリのインストール

sudo apt install wget git python3 python3-venv
sudo apt-get install libgoogle-perftools4 libtcmalloc-minimal4

Web UIのレポジトリをクローン

git clone --depth=1 --branch v1.4.1 https://github.com/AUTOMATIC1111/stable-diffusion-webui

インストール実行

./webui.sh

webui.shの中ではvenv上に自動でツール群をインストールする模様。デフォルトではtorch-2.0.1+cu118が入った。モデルはv1-5-pruned-emaonly.safetensorsというモデルがダウンロードされた (Stable Diffusion v1.5のPrunedバージョン？約4GB)。

Web UI起動

ウェブブラウザでhttp://127.0.0.1:7860を開いて下記のようなUIが表示されればOK。

動作確認

試しにプロンプト(Positive prompt)に "A photo of astronaut riding a horse" と入れてGenerateをクリック。画像が生成されればOK。

RTX 2070 SUPERで1枚生成するのに3秒くらい (512x512 pix, 20 iterations)