AWS EC2 に stable-diffusion-webuiをインストール方法です。
AMIの選択
CUDAの実行にNVIDIAのドライバーのインストールなどが必要です。素のAmazonLinux2を使うとセットアップが非常に面倒くさいので、ドライバーインストール済みのAMIを使います。
「Deep Learning AMI GPU PyTorch 1.13.1 (Amazon Linux 2) 20230201」 を選択します。
インスタンス
動作確認では g5.2xlarge を使ってます。GPUが搭載されたインスタンスを選択してください。
GPUが搭載されたインスタンスは非常に料金が高いので料金をよく確認して選択してください。
なお、安いインスタンスで環境を作ってから後で良いGPUを積んだインスタンスタイプに変更した方が経済的です。
起動後の確認
nvidia-smi コマンドを実行してGPUを確認する。
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 515.65.01 Driver Version: 515.65.01 CUDA Version: 11.7 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 NVIDIA A10G On | 00000000:00:1E.0 Off | 0 |
| 0% 35C P0 62W / 300W | 3505MiB / 23028MiB | 0% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 5897 C python3 3503MiB |
+-----------------------------------------------------------------------------+
python3.8のインストール
デフォルトではPython3.7がインストールされているが、stable-diffusion-webuiにはPython3.7では動かない依存があるため、Python3.8をインストールする。
sudo yum remove python3
sudo amazon-linux-extras enable python3.8
sudo yum install python38
cd /usr/bin/
sudo ln -s python3.8 python3
zlib1.2.9のインストール (非常に危険な作業です)
zlib1.2.9が要求されるのでインストールする。
失敗するとサーバーがまるごと壊れて復旧不可能になる 超危険 な作業です。壊れたら困る環境では実施しないでください。
curl -L -O https://sourceforge.net/projects/libpng/files/zlib/1.2.9/zlib-1.2.9.tar.gz/download
tar zxf download
cd zlib-1.2.9/
./configure
make
sudo mv libz.so.1.2.9 /lib64/
cd /lib64/
sudo ln -s -f libz.so.1.2.9 libz.so.1
stable-diffusion-webuiのインストール
stable-diffusion-webuiのインストールと実行
bash <(wget -qO- https://raw.githubusercontent.com/AUTOMATIC1111/stable-diffusion-webui/master/webui.sh)
下記のようなエラーが出た場合は、モデルデータをダウンロードして models/Stable-diffusion ディレクトリに保存する。
No checkpoints found. When searching for checkpoints, looked at:
StableDiffusion1.5のモデルデータのダウンロード。完了後に webui.sh を再度実行する。
cd /home/$(whoami)/stable-diffusion-webui/models/Stable-diffusion
wget -L https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned.safetensors
外部から接続する場合は、下記の環境変数を指定してから webui.sh を実行します。ただし、0.0.0.0にバインドされるので、7860ポートを外部に公開していると誰でもアクセスできしまうので注意が必要です。
export COMMANDLINE_ARGS="--listen"
Apple Silliconと比べた速度
Apple M1 MaxのMacBook Proではtxt2imgで20stepの時に10秒くらいかかるのが、g5.2xlargeでは2秒強くらいで完了します。さすが早いです。