1. Stable Diffusion中文网首页
  2. Stable Diffusion
  3. 模型

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

SD中文网 • • 模型 • 阅读 769

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

9b45b2d02b270d335cfaf2818fc5c6ac环境搭建

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

GitHub – bmaltais/kohya_ss

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

  1. 以管理员模式启动Powershell，并执行以下命令：
Set-ExecutionPolicy Unrestricted
​

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

  1. 启动一个普通的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
​
  1. 安装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
​
  1. 将必要的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的镜像。

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

原创文章，作者：SD中文网，如若转载，请注明出处：https://www.stablediffusion-cn.com/sd/sd-model/570.html

赞 (0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
SD中文网
上一篇 2023年8月16日 am8:00
下一篇 2023年8月17日 am8:00

相关推荐

加入交流群