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:
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

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:
    • Se não encontrado, execute:

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:

Então instale o AI Training:

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.
Você verá (aitraining-env) no seu terminal quando o ambiente estiver ativo.

Passo 2: Instalar o AI Training

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:
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:
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:

Configuração Weights & Biases (Opcional)

O AITraining usa W&B para logging por padrão. Para habilitar:
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:
Ou melhor, use um ambiente virtual (mostrado acima).
pip não foi instalado com Python. Instale:
Ou no Linux:
  • 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?