在Windows系统上安装AI Toolkit需要遵循以下关键步骤:
- armazém de clones:在命令提示符中执行
git clone https://github.com/ostris/ai-toolkit.git
Em seguida, vá para o diretório do projetocd ai-toolkit
- Atualização de submódulos: Executar
git submodule update --init --recursive
确保所有依赖库完整 - Criação de um ambiente virtualUse o comando
python -m venv venv
创建隔离环境,并通过venvScriptsactivate
ativar - Instalação de dependências:需要特别注意:
- 先安装特定版本的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加速功能正常工作的关键。
Essa resposta foi extraída do artigoKit de ferramentas de IA da Ostris: Difusão estável com o kit de ferramentas de treinamento de modelos FLUX.1O