Saltar al contenido principal

​
Instalación y Configuración

El CLI de AITraining está incluido cuando instalas el paquete aitraining.

​
Instalación

​
Usando pip

pip install aitraining torch torchvision torchaudio

​
Usando uv (Recomendado)

uv pip install aitraining torch torchvision torchaudio

​
Verificar Instalación

Verifica que el CLI funciona: 
aitraining --version
Deberías ver el número de versión impreso.

​
Primera Ejecución

Ejecutar aitraining sin argumentos inicia el asistente interactivo: 
aitraining
Esto muestra el banner de Monostate e inicia el asistente de configuración.

​
Configuración de GPU

Para entrenamiento con GPU, asegúrate de tener CUDA instalado y que PyTorch pueda detectar tu GPU: 
import torch
print(torch.cuda.is_available())  # Should print True
print(torch.cuda.device_count())  # Number of GPUs

​
Apple Silicon (M1/M2/M3)

En Macs Apple Silicon, MPS se usa automáticamente: 
import torch
print(torch.backends.mps.is_available())  # Should print True

​
Variables de Entorno

Variables de entorno comunes:
VariableDescripción
HF_TOKENToken de Hugging Face para modelos/datasets privados
WANDB_API_KEYClave API de Weights & Biases para logging
CUDA_VISIBLE_DEVICESEspecificar qué GPUs usar
Configura estas antes de ejecutar el entrenamiento: 
export HF_TOKEN="your_token_here"
export WANDB_API_KEY="your_key_here"

​
Próximos Pasos

Estructura de Comandos

Aprende la sintaxis de comandos CLI

Entrenamiento LLM

Entrena modelos de lenguaje