記事概要
話題の「GitHub - CompVis/stable-diffusion」で遊んでみる
日本語の解説記事とかも充実しているので参考にしたサイトをメモするだけになりそう
環境構築
PC環境
- Ubuntu 20.04 on WSL2 (Windows 10)
- Intel(R) Core(TM) i5-11400
- RAM32GB(WSL2に16GB)
- NVIDIA GeForce RTX 3060(VRAM 12GB)
参考記事
画像生成AI「Stable Diffusion」をローカル環境で実行する - パソコン関連もろもろ
WSLからGPUを使えるようにする
上記記事を参考に構築して実行してみると
RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx
と出た。
調べたところWindows10の場合はOS versionが21H2でないとWSL上でGPUを使えない模様。
(winverで調べたところ21H2だった)
参考: Enable NVIDIA CUDA on WSL 2 | Microsoft Docs
諸般の事情でまだWindows11にしたくないので、書いてある通り21H2をインストール。
蛇足
ぐぐるとNVIDIA公式が出てきてWindows Insider Programに参加して適切なOS buildにすると書いてあるが、
今それをやるとWindows11に誘導されるだけなので、2022/8現在はMicrosoftに書いてある通り21H2にするのが良さそう
実行
sampleのを実行してみる
$ python3 scripts/txt2img.py --prompt "a photograph of an astronaut riding a horse" --plms --ckpt sd-v1-4.ckpt --n_samples 1
Your samples are ready and waiting for you here: outputs/txt2img-samples Enjoy.
やったぜ。