在Windows系统上安装AI Toolkit需要遵循以下关键步骤:
- 克隆仓库:在命令提示符中执行
git clone https://github.com/ostris/ai-toolkit.git
,然后进入项目目录cd ai-toolkit
- 更新子模块:运行
git submodule update --init --recursive
确保所有依赖库完整 - 创建虚拟环境:使用命令
python -m venv venv
创建隔离环境,并通过venvScriptsactivate
激活 - 安装依赖:需要特别注意:
- 先安装特定版本的PyTorch(支持CUDA 12.4):
pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cu124
- 再安装其他依赖项:
pip install -r requirements.txt
- 先安装特定版本的PyTorch(支持CUDA 12.4):
安装完成后,用户可根据需要通过Gradio UI或命令行界面运行工具集。相比于Linux版本,Windows安装需要特别注意CUDA版本匹配问题,这是保证GPU加速功能正常工作的关键。
本答案来源于文章《AI Toolkit by Ostris:Stable Diffusion与FLUX.1模型训练工具包》