在Windows系统上安装AI Toolkit需要遵循以下关键步骤:
- clone warehouse:在命令提示符中执行
git clone https://github.com/ostris/ai-toolkit.git
Then go to the project directorycd ai-toolkit
- Updating submodules: Run
git submodule update --init --recursive
确保所有依赖库完整 - Creating a Virtual Environment: Use the command
python -m venv venv
创建隔离环境,并通过venvScriptsactivate
activation - Installation of dependencies:需要特别注意:
- 先安装特定版本的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加速功能正常工作的关键。
This answer comes from the articleAI Toolkit by Ostris: Stable Diffusion with FLUX.1 Model Training ToolkitThe