Nota
El acceso a esta página requiere autorización. Puede intentar iniciar sesión o cambiar directorios.
El acceso a esta página requiere autorización. Puede intentar cambiar los directorios.
La databricks-ai-runtime habilidad de agente proporciona a los agentes de programación de IA, como Claude Code, Cursor y Codex, instrucciones para trabajar con AI Runtime a través de los databricks air comandos de la CLI de Databricks. Con la habilidad instalada, puede pedirle a su agente que escriba un archivo YAML de carga de trabajo, envíe una ejecución de entrenamiento en GPU y compruebe su progreso.
Requirements
- La CLI de Databricks, autenticada en tu espacio de trabajo. Los ejemplos de la habilidad utilizan el formato
code_sourceaplanado, que requiere la versión 1.20.0 o superior de la CLI de Databricks. - Un espacio de trabajo con AI Runtime activado. Ver Requisitos.
Instalar la habilidad
Ejecute el siguiente comando:
databricks aitools install --skills-only --skills databricks-ai-runtime
Sigue las indicaciones para elegir tu agente de IA y el alcance de la instalación. Para opciones de instalación, consulta aitools el grupo de comandos.
Qué abarca esta habilidad
| Capacidad | Qué puede hacer el agente |
|---|---|
| Escribir cargas de trabajo | Convierte una solicitud en un YAML de carga de trabajo con computación, entorno, código fuente y el comando de lanzamiento de tu framework, como torchrun, Accelerate o DeepSpeed. |
| Configurar ejecuciones | Establece variables de entorno, secretos, hiperparámetros, reintentos y permisos. |
| Enviar ejecuciones | Valida la configuración con --dry-run, y luego envía con databricks air run y una clave de idempotencia. |
| Revisa y monitoriza las ejecuciones | Obtener el estado con databricks air get y databricks air list, transmitir registros con databricks air logs y remitirte a la ejecución de MLflow para consultar las métricas. |
| Cancelar ejecuciones | Cancela una, varias o todas tus ejecuciones activas con databricks air cancel. |
| Fallos de depuración | Saca los registros de una ejecución fallida y muestra el error. |
| Uso de imágenes personalizadas de Docker | Compilar y enviar una imagen personalizada y, a continuación, hacer referencia a ella en la carga de trabajo. |