Pular para o conteúdo principal

​
Instalando o AI Training

Este guia vai te orientar na instalação do AI Training no seu computador. O processo leva cerca de 5-10 minutos.

​
O Que Você Vai Obter

Quando você instala o AI Training com pip ou uv, você obtém:
  • Interface de Linha de Comando - O comando aitraining para treinar modelos
  • Interface de Chat - Teste e interaja com seus modelos treinados via navegador
  • API Python - Importe e use em seus scripts Python
Todas as interfaces estão incluídas em uma única instalação de pacote.

​
Pré-requisitos

Antes de instalar o AI Training, você precisa do Python no seu sistema. Python é uma linguagem de programação na qual o AI Training executa.

​
Verificar se o Python Está Instalado

Abra seu terminal (Mac) ou Prompt de Comando (Windows) e digite: 
python --version
Se você ver um número de versão como Python 3.8.0 ou superior, está tudo certo. O AI Training requer Python 3.8 ou mais recente (3.10+ recomendado). Se você receber um erro ou tiver uma versão mais antiga, siga os passos de instalação abaixo.

​
Instalando o Python

  • Windows
  • Mac
  • Linux

​
Instalação no Windows

  1. Baixar Python
  2. Executar o Instalador
    • Clique duas vezes no arquivo baixado
    • Importante: Marque “Add Python to PATH” na parte inferior
    • Clique em “Install Now”
    • Aguarde a instalação completar
  3. Verificar Instalação
    • Abra o Prompt de Comando (procure por “cmd” no menu Iniciar)
    • Digite: python --version
    • Você deve ver a versão do Python
  4. Instalar pip (se necessário)
    • pip geralmente vem com Python, mas verifique:
    pip --version
    • Se não encontrado, execute:
    python -m ensurepip --upgrade

​
Instalando o AI Training

Uma vez que o Python está pronto, você tem duas opções para instalar o AI Training. Recomendamos usar uv para instalação mais rápida.

​
Opção 1: Usando uv (Recomendado)

uv é um instalador de pacotes Python moderno que é muito mais rápido que pip e gerencia ambientes virtuais automaticamente.

​
Instale uv primeiro:

  • Windows
  • Mac/Linux
# Using PowerShell (as administrator)
powershell -c "irm https://astral.sh/uv/install.ps1 | iex"

​
Então instale o AI Training:

# uv cria e gerencia automaticamente um ambiente virtual
uv pip install aitraining
Pronto! O uv gerencia o ambiente virtual para você automaticamente.

​
Opção 2: Usando pip (Método tradicional)

Se você prefere a abordagem tradicional ou não consegue instalar uv:

​
Passo 1: Criar um Ambiente Virtual

Um ambiente virtual mantém os arquivos do AI Training separados de outros projetos Python.
  • Windows
  • Mac/Linux
python -m venv aitraining-env
aitraining-env\Scripts\activate
Você verá (aitraining-env) no seu terminal quando o ambiente estiver ativo.

​
Passo 2: Instalar o AI Training

pip install aitraining
Isso baixa e instala o AI Training do PyPI com todas as suas dependências (incluindo PyTorch).

​
Passo 3: Verificar Instalação

Verifique se tudo foi instalado corretamente: 
aitraining --version
Você deve ver o número da versão do AI Training.

​
Escolha Sua Interface

O AI Training oferece três formas de trabalhar:

Linha de Comando

Melhor para treinamentoTreine modelos com comandos. Bom para scripts e automação.Comece com: aitraining --help

Interface de Chat

Melhor para testesTeste seus modelos treinados interativamente em um navegador.Comece com: aitraining chat

API Python

Melhor para desenvolvedoresIntegre treinamento em suas aplicações Python.Comece com: from autotrain.trainers.clm.params import LLMTrainingParams

​
Dependências Opcionais

Alguns recursos requerem pacotes adicionais:
RecursoPacoteInstalarPlataforma
Treinamento LoRA mais rápidounslothpip install unslothLinux (apenas SFT, llama/mistral/gemma/qwen2)
Flash Attention 2flash-attnpip install flash-attnLinux (CUDA)
Treinamento distribuídoaccelerateIncluídoLinux/Mac/Windows
DeepSpeed Zero-3deepspeedpip install deepspeedLinux
Quantização 4/8-bitbitsandbytesIncluídoApenas Linux
Accelerate é necessário para treinamento multi-GPU e distribuído. Está incluído com o AITraining. DeepSpeed é opcional e fornece otimização de memória Zero-3 para modelos muito grandes.
Quantização (int4/int8) funciona apenas no Linux. A biblioteca bitsandbytes necessária para --quantization int4 ou --quantization int8 não está disponível no Windows ou macOS.

​
Configuração de GPU (Opcional)

Para treinamento com GPU NVIDIA, você pode querer reinstalar PyTorch com sua versão específica de CUDA: 
# CUDA 11.8
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

# CUDA 12.1
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

​
Configuração Weights & Biases (Opcional)

O AITraining usa W&B para logging por padrão. Para habilitar: 
pip install wandb
wandb login
Ou defina a variável de ambiente: WANDB_API_KEY=your_key

​
Problemas Comuns

Python não está no PATH do seu sistema. Tente:
  • Use python3 em vez de python
  • Reinstale Python e marque “Add Python to PATH”
  • Adicione Python manualmente ao PATH nas Variáveis de Ambiente do Sistema
No Mac/Linux, você pode precisar usar sudo:
sudo pip install aitraining
Ou melhor, use um ambiente virtual (mostrado acima).
pip não foi instalado com Python. Instale:
python -m ensurepip --upgrade
Ou no Linux:
sudo apt install python3-pip
  • Certifique-se de ter espaço em disco suficiente (PyTorch é grande)
  • Tente atualizar pip primeiro: pip install --upgrade pip
  • Para suporte a GPU, certifique-se de que os drivers CUDA estão instalados

​
Próximos Passos

Instalação completa! Agora você está pronto para treinar seu primeiro modelo:

Tutorial de Início Rápido

Treine seu primeiro modelo de IA em 10 minutos

​
Precisa de Ajuda?