在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ツールキット:FLUX.1モデルトレーニングツールキットによる安定した拡散について