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 agentes de codificació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, puedes pedirle a tu agente que escriba un YAML de carga de trabajo, envíe una ejecución de entrenamiento en GPU y compruebe su progreso.
Requirements
- La CLI de Databricks está autenticada en tu espacio de trabajo. Los ejemplos del comando utilizan el formato aplanado
code_source, 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.
Instala la skill
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é incluye esta skill
| Capacidad | Qué puede hacer el agente |
|---|---|
| cargas de trabajo de escritura | 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 las 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 | Obtén estado con databricks air get y databricks air list, transmite los registros en tiempo real con databricks air logs, y te redirige a la ejecución de MLflow para ver 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 | Construye y sube una imagen personalizada, y luego haz referencia a ella en la carga de trabajo. |