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:

Variable	Descripción
`HF_TOKEN`	Token de Hugging Face para modelos/datasets privados
`WANDB_API_KEY`	Clave API de Weights & Biases para logging
`CUDA_VISIBLE_DEVICES`	Especificar qué GPUs usar

Configura estas antes de ejecutar el entrenamiento:

export HF_TOKEN="your_token_here"
export WANDB_API_KEY="your_key_here"

Básicos de CLI

Configuración

Comandos de Entrenamiento

Uso Avanzado

Inferencia

Instalación y Configuración

Instalación y Configuración

Instalación

Usando pip

Usando uv (Recomendado)

Verificar Instalación

Primera Ejecución

Configuración de GPU

Apple Silicon (M1/M2/M3)

Variables de Entorno

Próximos Pasos

Estructura de Comandos

Entrenamiento LLM

​Instalación y Configuración

​Instalación

​Usando pip

​Usando uv (Recomendado)

​Verificar Instalación

​Primera Ejecución

​Configuración de GPU

​Apple Silicon (M1/M2/M3)

​Variables de Entorno

​Próximos Pasos

Estructura de Comandos

Entrenamiento LLM

Instalación y Configuración

Instalación

Usando pip

Usando uv (Recomendado)

Verificar Instalación

Primera Ejecución

Configuración de GPU

Apple Silicon (M1/M2/M3)

Variables de Entorno

Próximos Pasos