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

时间:23-10-13 15:21:13

作者:

分类:AI软件教程

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

9b45b2d02b270d335cfaf2818fc5c6ac环境搭建

在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的镜像。

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

最新推荐