一. 背景

上期給大家介紹了如何用騰訊云服務(wù)器快速部署chilloutmix

但簡(jiǎn)單的模型部署只是千里之行的第一步。有人把Stable Diffusion比喻成單反，Midjourney比喻成數(shù)字相機(jī)，感覺(jué)非常貼切。想玩好單反，各種參數(shù)，插件的熟練應(yīng)用是必不可缺的。

這兩天發(fā)現(xiàn)騰訊云的GPU實(shí)驗(yàn)室支持了預(yù)裝Docker的容器鏡像，廢話不多說(shuō)，本次給大家?guī)?lái)更簡(jiǎn)潔的一鍵安裝AI繪畫(huà)集成包操作指南。

二. 實(shí)現(xiàn)效果

安裝完成后你將擁有一個(gè)這樣的環(huán)境：預(yù)裝SD-v1_5模型，預(yù)裝Dreambooth插件（訓(xùn)練），預(yù)裝controlnet插件（控圖），預(yù)裝jupyternotebook并默認(rèn)啟動(dòng)（可視化文件管理）

安裝共需：兩步

安裝耗時(shí)約：看網(wǎng)速，最快20min搞定

三. 操作步驟

3.1 買(mǎi)一臺(tái)服務(wù)器（1min）

首先需要購(gòu)買(mǎi)一臺(tái)騰訊云的GPU服務(wù)器。GPU卡型本次選擇“GN10X”，卡型對(duì)應(yīng)“V100”，性能比“GN7”強(qiáng)不少（另一個(gè)原因是GN7被搶完了，最近太火爆）。我是從GPU實(shí)驗(yàn)室活動(dòng)頁(yè)搶購(gòu)到的折扣機(jī)型（https://cloud.tencent.com/act/pro/gpu-study）

注意，點(diǎn)擊立即購(gòu)買(mǎi)后，預(yù)裝鏡像要選擇“Docker基礎(chǔ)鏡像”。若選擇其他鏡像，可能會(huì)導(dǎo)致無(wú)法順利跑通本流程。

3.2 拉取鏡像（10-15min）

登錄機(jī)器后，直接輸入如下命令，拉取容器鏡像。拉取約需10-15min，可以先喝杯茶，休息一下。

sudo docker pull gpulab.tencentcloudcr.com/ai/stable-diffusion:1.0.6

3.3 啟動(dòng)容器，完成部署（1min）

1. 復(fù)制粘貼下述命令，啟動(dòng)容器，完成部署

sudo docker run -itd --gpus=all --network=host --device=/dev/dri --group-add=video --ipc=host --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --name=stable-diffusion gpulab.tencentcloudcr.com/ai/stable-diffusion:1.0.6 | xargs sudo docker logs --follow

1. 使用生成的public url，即可啟動(dòng)AI繪畫(huà)環(huán)境

四. 一些簡(jiǎn)單的使用介紹

4.1 使用容器

a. 進(jìn)入容器

sudo docker exec -it stable-diffusion bash

b. 退出容器

exit

c. 重啟容器

sudo docker restart stable-diffusion | xargs sudo docker logs --follow

4.2 使用jupternotebook

啟動(dòng)容器過(guò)程中，會(huì)展示jupyternotebook的登錄鏈接，只需要復(fù)制url，將“127.0.0.1”替換為本機(jī)實(shí)例的公網(wǎng)ip，粘貼至瀏覽器后，即可啟動(dòng)jupyternotebook。下面是樣例：

替換前：http://127.0.0.1:8888/?token=0b5bb7c181787a36964df086ef7d90b392ded4ff001b

替換后：http://1.116.111.48:8888/?token=0b5bb7c181787a36964df086ef7d90b392ded4ff001b

4.3 使用controlnet

a. 由于controlnet的模型種類(lèi)較多，體積較大，這個(gè)環(huán)境沒(méi)有提前預(yù)裝任何controlnet模型，僅預(yù)裝了插件。

b. 若需要使用controlnet的能力，需前往https://huggingface.co/webui/ControlNet-modules-safetensors/tree/main下載需要使用的模型，并保存至容器中的"stable-diffusion-webui/extensions/sd-webui-controlnet/models"目錄下

4.4 使用dreambooth

a. 通過(guò)本預(yù)裝插件，可快速進(jìn)行模型訓(xùn)練。具體訓(xùn)練方式，參數(shù)配置等信息可參考網(wǎng)絡(luò)上的教程