Stable Diffusion训练模型环境部署（Linux操作系统上部署GUI）

　　作为一个机器学习研究人员或开发者，环境搭建对于进行模型训练是非常重要的一步。本文将以Stable Diffusion为例，介绍如何在Windows操作系统下搭建模型训练环境，并提供Linux上的GUI部署参考。

环境搭建

在Windows操作系统下搭建模型训练环境

GitHub – bmaltais/kohya_ss

　　首先，我们以GitHub上的bmaltais/kohya_ss项目为例，该项目提供了在Windows操作系统下的GUI训练面板。在进行环境搭建之前，请确保设备已经安装了Python 3.10.6和git环境。

　　以管理员模式启动Powershell，并执行以下命令：

Set-ExecutionPolicy Unrestricted
​

　　在提示时回答”A”，然后可以关闭该窗口。

　　启动一个普通的Powershell窗口，并进入需要克隆该仓库的路径下，执行以下命令：

git clone https://github.com/bmaltais/kohya_ss.git
cd kohya_ss
python -m venv venv
.\venv\Scripts\activate
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
pip install --use-pep517 --upgrade -r requirements.txt
​

　　安装GUI所需的依赖库：

pip install -U -I --no-deps https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl
​

　　将必要的dll文件复制到虚拟环境中：

cp .\bitsandbytes_windows\*.dll .\venv\Lib\site-packages\bitsandbytes\
​

　　通过以上步骤，您已经成功搭建了在Windows操作系统下使用GUI进行模型训练的环境。

在Linux操作系统上部署GUI

　　如果您需要在Linux上部署，并且需要GUI界面，我们可以参考GitHub上的P2Enjoy/kohya_ss-docker项目。该项目是将kohya_ss训练Web UI转换为适用于Linux的镜像。

　　根据这个项目的说明进行部署即可。