![images](../img/imgtop_sub_b.png)
拡散モデルの操作を、CLIベースで行えるツールです。
関連
- ◯
- クラウドのコンテナでGPUを使う:GPC, Docker, nvidia-driver, cuda-toolkit
- ◯
- コンテナの各種メディアを、遠隔からあつかう(音声・画像・映像):Docker, ssh, rsync, sshfs, nginx, Tk (tkinter) , OpenCV, Jupyter Notebook
検証
- ・
- OS: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}