はじめに
今年になって悪ノリでRTX 4090を購入したもののゲーム熱も冷めてきて何かに使えないかなーってなってました。
どうせだしお絵描きAIで遊ぶかーと思い立ってstable-diffusion-webui-dockerを起動しようとしたもののすんなりいかなかったのでその際の備忘を
問題
「docker compose --profile auto up --build」をたたくと下記のエラーが出る
Attaching to webui-docker-auto-1
Error response from daemon: could not select device driver "nvidia" with capabilities: [[compute utility]]
解決方法
シンプルに「nvidia-container-runtime」を入れなおせばよかった
NVIDIAの公式リポジトリ追加して～
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$(. /etc/os-release;echo $ID$VERSION_ID)/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
aptのパッケージ更新して～
sudo apt-get update
「nvidia-container-runtime」を入れなおして～
sudo apt-get install nvidia-container-runtime
念のためDocker再起動して～
sudo systemctl restart docker
build再実行
docker compose --profile auto up --build
おわりに
適当にサイバーパンクな風景やってみるかーで試してみたところ
画像出力2-3秒くらいでぽいぽいできるので楽しいかもです
参考