Tavris1/AI-Toolkit-Easy-Install

解决的问题

搭建扩散模型训练环境是出了名的麻烦：你需要匹配版本的 Python、PyTorch、CUDA、FlashAttention、FFmpeg 等。本项目将这些依赖打包到一个可下载的 ZIP 文件中，其中包含一个 .bat 文件。运行它会安装或更新所有必需的内容，并启动 AI-Toolkit 的 GUI/本地服务器，而不会影响你的系统 Python 或创建虚拟环境。

工作原理

你下载发布版 ZIP，将批处理文件解压到一个新文件夹中，然后双击运行。安装程序会检查并安装 Git、Node.js、FFmpeg、嵌入式 Python 3.12.10、带 CUDA 13.0 的 PyTorch 2.13.0、Triton、FlashAttention、torchcodec、huggingface_hub 扩展，最后安装最新的 AI-Toolkit 本身。准备就绪后，启动器会启动服务器，在服务器上线时打开你的默认浏览器，并检查更新。一个单独的 Update-AI-Toolkit 脚本负责后续升级。

适用人群

适用于拥有 NVIDIA GPU 的 Windows 用户，他们希望训练或微调图像和视频扩散模型，而无需手动处理包管理器、CUDA 版本或虚拟环境。它面向初学者和有经验的用户，他们更喜欢便携、自包含的设置。

亮点

• 一键便携安装器 – 无需系统 Python，无需虚拟环境，只需运行批处理文件。
• 自动配置整个技术栈 – Git、Node.js、FFmpeg、Python、PyTorch、Triton、FlashAttention、torchcodec 以及 AI-Toolkit 本身。
• 固定兼容版本 – FlashAttention 2.8.3、Triton 3.7.1.post27、torchcodec 0.15.0（用于 Torch 2.13.0）以避免破坏。
• 启动器便利性 – 服务器启动时打开浏览器，并提醒你更新。
• 安全使用指南 – 警告不要以管理员身份运行、不要在系统文件夹中运行、路径中不要有空格。