Nota
O acesso a esta página requer autorização. Pode tentar iniciar sessão ou alterar os diretórios.
O acesso a esta página requer autorização. Pode tentar alterar os diretórios.
A databricks-ai-runtime skill de agente fornece aos agentes de programação de IA, como Claude Code, Cursor e Codex, instruções para trabalhar com o Runtime de IA através dos databricks air comandos CLI do Databricks. Com a skill instalada, pode pedir ao seu agente para escrever um YAML de carga de trabalho, submeter uma execução de treino da GPU e verificar o seu progresso.
Requirements
- A linha de comandos do Databricks, autenticada no seu espaço de trabalho. Os exemplos da skill utilizam o formato achatado
code_source, que requer a CLI Databricks versão 1.20.0 ou superior. - Um espaço de trabalho com AI Runtime ativado. Ver Requisitos.
Instala a habilidade
Execute o seguinte comando:
databricks aitools install --skills-only --skills databricks-ai-runtime
Siga as instruções para escolher o seu agente de IA e o âmbito da instalação. Para opções de instalação, consulte aitools o grupo de comandos.
O que a competência cobre
| Capacidade | O que o agente pode fazer |
|---|---|
| Cargas de trabalho de gravação | Transforme uma solicitação num YAML de carga de trabalho com computação, ambiente, código fonte e o comando de lançamento do seu framework, como torchrun, Accelerate ou DeepSpeed. |
| Configurar execuções | Defina variáveis de ambiente, segredos, hiperparâmetros, novas tentativas e permissões. |
| Submeter execuções | Valide a configuração com --dry-run, depois submeta com databricks air run e uma chave de idempotência. |
| Verificar e monitorizar as execuções | Obtenha o estado com databricks air get e databricks air list, veja os registos em fluxo com databricks air logs e encaminhe-o para a execução do MLflow para ver as métricas. |
| Cancelar execuções | Cancela uma, várias ou todas as tuas execuções ativas com databricks air cancel. |
| Falhas de depuração | Extraia os registos de uma execução falhada e apresenta o erro. |
| Use imagens Docker personalizadas | Constrói e envia uma imagem personalizada, depois referencia-a no workload. |