Windows下运行AutoTrain DreamBooth遇Triton依赖问题求解决方案

方案1：禁用Triton依赖（无需切换环境） #

Triton为Linux专属模块，可通过设置环境变量让diffusers跳过Triton加载：

• 执行训练命令前，先在命令行设置环境变量：

  set DISABLE_TRITON=1
  

• 随后重新运行你的AutoTrain命令：

  autotrain dreambooth --model stabilityai/stable-diffusion-xl-base-1.0 --image-path ./Anas --prompt 'a photo of arzk man' --resolution 1024 --batch-size 1 --num-steps 500 --fp16 --gradient-accumulation 4 --lr 1e-4 --project-name thesis
  

若仍报错，建议安装兼容Windows的diffusers版本：

pip install diffusers==0.21.4

方案2：用WSL2在Windows上运行Linux环境（无需完全切换系统） #

WSL2是Windows自带的轻量Linux子系统，操作门槛低：

1. 打开Windows设置，进入「应用」→「可选功能」→「添加功能」，搜索并安装「适用于Linux的Windows子系统」
2. 打开Microsoft Store，搜索Ubuntu并安装（优先选LTS稳定版本）
3. 启动Ubuntu，设置用户名和密码后，更新系统：

   sudo apt update && sudo apt upgrade -y
   

4. 安装Python和pip：

   sudo apt install python3 python3-pip -y
   

5. 安装autotrain-cli：

   pip install autotrain-cli
   

6. 将本地./Anas图片文件夹复制到WSL的用户目录（例如/home/你的用户名/Anas），再运行训练命令即可。

方案3：使用在线GPU平台（无需本地配置） #

借助Google Colab的免费GPU额度即可完成训练：

1. 打开Colab新建Python笔记本
2. 执行安装命令：

   !pip install autotrain-cli
   

3. 点击左侧「文件」图标，上传你的Anas图片文件夹
4. 在笔记本中运行训练命令（注意将--image-path调整为上传后的路径，如./Anas）
5. 训练完成后，下载生成的模型文件即可。

内容的提问来源于stack exchange，提问作者Anas Rzq