As etapas principais a seguir precisam ser seguidas para instalar o AI Toolkit em um sistema Windows:
- armazém de clonesExecutar no prompt de comando
git clone https://github.com/ostris/ai-toolkit.gitEm seguida, vá para o diretório do projetocd ai-toolkit - Atualização de submódulos: Executar
git submodule update --init --recursiveAssegure-se de que todas as bibliotecas dependentes estejam completas - Criação de um ambiente virtualUse o comando
python -m venv venvCrie um ambiente isolado e passe o arquivovenvScriptsactivateativar - Instalação de dependênciasAtenção especial é necessária:
- Instale primeiro uma versão específica do PyTorch (o CUDA 12.4 é compatível):
pip install torch==2.5.1 torchvision==0.20.1 --index-url https://download.pytorch.org/whl/cu124 - Instale as outras dependências novamente:
pip install -r requirements.txt
- Instale primeiro uma versão específica do PyTorch (o CUDA 12.4 é compatível):
Após a conclusão da instalação, os usuários podem executar o conjunto de ferramentas por meio da interface do usuário do Gradio ou da interface de linha de comando, conforme necessário. Em comparação com a versão Linux, a instalação do Windows requer atenção especial à correspondência da versão CUDA, que é crucial para garantir que os recursos de aceleração de GPU funcionem corretamente.
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






























