拡散モデルの操作を、ＣＬＩベースで行えるツールです。
関連
- ◯
- クラウドのコンテナでＧＰＵを使う：GPC, Docker, nvidia-driver, cuda-toolkit
- ◯
- コンテナの各種メディアを、遠隔からあつかう（音声・画像・映像）：Docker, ssh, rsync, sshfs, nginx, Tk (tkinter) , OpenCV, Jupyter Notebook
検証
- ・
- ＯＳ：Ubuntu 22.04 (Docker Hub)
- ・
- インタプリタ：Python 3.10.6
設置
リモート側で、必要なファイルをダウンロードします：
$ cd ${dir}
$ git clone --depth 1 https://github.com/kohya-ss/sd-scripts.git
コンテナの初期設定ファイルは、たとえば次のように記述します：
FROM ubuntu:22.04 RUN apt -y update RUN apt -y upgrade
コンテナを作成します：
$ docker build --no-cache -t ${image} ${dockerfile_directory}
コンテナを起動します：
$ docker run -it --rm \
-v ${dir}:${dir} \
--gpus all \
--name ${container} \
${image}
コンテナ上で、次を実行します：
$ apt install python3=3.10.6-1~22.04
$ apt install pip
$ pip install torch
$ cd ${dir}/sd-scripts
$ pip install --upgrade -r requirements.txt
$ accelerate config
> 0,0,ENTER,ENTER,ENTER,ENTER,0
$ accelerate test
リモート側で、必要に応じてコミットします：
$ docker commit ${container} ${image}