上期给大家介绍了如何用腾讯云服务器快速部署chilloutmix
但简单的模型部署只是千里之行的第一步。有人把Stable Diffusion比喻成单反,Midjourney比喻成数字相机,感觉非常贴切。想玩好单反,各种参数,插件的熟练应用是必不可缺的。
这两天发现腾讯云的GPU实验室支持了预装Docker的容器镜像,废话不多说,本次给大家带来更简洁的一键安装AI绘画集成包操作指南。
安装完成后你将拥有一个这样的环境:预装SD-v1_5模型,预装Dreambooth插件(训练),预装controlnet插件(控图),预装jupyternotebook并默认启动(可视化文件管理)
安装共需:两步
安装耗时约:看网速,最快20min搞定
首先需要购买一台腾讯云的GPU服务器。GPU卡型本次选择“GN10X”,卡型对应“V100”,性能比“GN7”强不少(另一个原因是GN7被抢完了,最近太火爆)。我是从GPU实验室活动页抢购到的折扣机型(https://cloud.tencent.com/act/pro/gpu-study)
注意,点击立即购买后,预装镜像要选择“Docker基础镜像”。若选择其他镜像,可能会导致无法顺利跑通本流程。
登录机器后,直接输入如下命令,拉取容器镜像。拉取约需10-15min,可以先喝杯茶,休息一下。
sudo docker pull gpulab.tencentcloudcr.com/ai/stable-diffusion:1.0.6
复制粘贴下述命令,启动容器,完成部署
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
使用生成的public url,即可启动AI绘画环境
a. 进入容器
sudo docker exec -it stable-diffusion bash
b. 退出容器
exit
c. 重启容器
sudo docker restart stable-diffusion | xargs sudo docker logs --follow
启动容器过程中,会展示jupyternotebook的登录链接,只需要复制url,将“127.0.0.1”替换为本机实例的公网ip,粘贴至浏览器后,即可启动jupyternotebook。下面是样例:
替换前:http://127.0.0.1:8888/?token=0b5bb7c181787a36964df086ef7d90b392ded4ff001b
替换后:http://1.116.111.48:8888/?token=0b5bb7c181787a36964df086ef7d90b392ded4ff001b
a. 由于controlnet的模型种类较多,体积较大,这个环境没有提前预装任何controlnet模型,仅预装了插件。
b. 若需要使用controlnet的能力,需前往https://huggingface.co/webui/ControlNet-modules-safetensors/tree/main下载需要使用的模型,并保存至容器中的"stable-diffusion-webui/extensions/sd-webui-controlnet/models"目录下
a. 通过本预装插件,可快速进行模型训练。具体训练方式,参数配置等信息可参考网络上的教程