air grupo de comando
O grupo de comandos air dentro da CLI do Databricks envia e gerencia cargas de trabalho de GPU, como treinamento e inferência, com o AI Runtime.
Para os campos de configuração de Workload, consulte Referência de YAML de Workload. Para enviar sua primeira carga de trabalho, consulte Início rápido do AI Runtime.
databricks air execução
Enviar uma carga de trabalho a partir de uma configuração YAML. A saída inclui o ID da execução do job e o URL de execução do job.
databricks air run --file YAML_PATH [flags]
Argumentos
Nenhuma
Opções
-f, --file string
Path to the workload YAML configuration. This option is required.
--watch
Transmissão dos logs do nó 0 até que a execução alcance um estado final. Default: false.
--override stringArray
Override a YAML field using FIELD_PATH=VALUE. Repeat this option to override multiple fields.
--dry-run
Valide a configuração sem enviar uma carga de trabalho. Default: false.
--idempotency-key string
Retorne uma execução existente quando a key tiver sido usada anteriormente. Substitui idempotency_token no YAML. Máximo: 64 caracteres.
Exemplos
O exemplo a seguir envia uma carga de trabalho de train.yaml:
databricks air run --file train.yaml
Passe um caminho de configuração como um argumento separado para -h para exibir a ajuda de configuração de carga de trabalho:
databricks air run -h config
databricks air run -h config.compute
databricks air run -h config.compute.accelerator_type
databricks air get
Obter detalhes de uma execução.
databricks air get JOB_RUN_ID [flags]
Argumentos
JOB_RUN_ID
A job execução ID returned by databricks air run.
Opções
databricks air list
Listar execuções. Por default, o comando lista as execuções ativas enviadas pelo usuário atual.
databricks air list [flags]
Argumentos
Nenhuma
Opções
--limit int
Número máximo de execuções a exibir. Default: 20.
--all-status
Incluir execuções em estados terminais. Default: false.
--all-users
Inclua execuções enviadas por outros usuários. Default: false.
--filter stringArray
Filtrar execuções usando KEY=VALUE. Repita esta opção para exigir várias condições. As chaves suportadas são accelerator_type, experiment, num_accelerators e user.
databricks air logs
Transmissão ou busca de logs para uma execução.
databricks air logs JOB_RUN_ID [flags]
Argumentos
JOB_RUN_ID
A job execução ID returned by databricks air run.
Opções
--node int
Selecione um índice de nó baseado em zero. Default: 0.
--tail int
Número de linhas de log existentes a serem impressas. Default: 10000.
--minutes int
Buscar logs dos últimos minutos especificados. Não pode ser combinado com --tail. Default: 0.
--retry int
Selecione uma tentativa de repetição. 0 é a tentativa inicial e -1 é a mais recente. Default: -1.
--download-to string
Faça o download dos logs completos para um diretório. Faz o download de todos os nós, a menos que --node esteja definido. Não é possível combinar com --tail ou --minutes.
databricks air cancel
Cancelar uma ou mais execuções pelo ID de execução do job ou cancelar todas as execuções ativas enviadas pelo usuário atual.
databricks air cancel [JOB_RUN_ID...] [flags]
Argumentos
JOB_RUN_ID
Um ou mais IDs de execução do job. Não pode ser combinado com --all.
Opções
--all
Cancelar todas as execuções ativas enviadas pelo usuário atual. Default: false.
-y, --yes
Skip the confirmation prompt used by --all. Default: false.
databricks air images push
Enviar uma imagem de contêiner para o Artifact Registry.
Para ver os pré-requisitos e as instruções para enviar uma imagem, consulte Push images with the Databricks CLI.
databricks air images push [flags]
Argumentos
Nenhuma
Opções
-s, --source string
A imagem de contêiner de origem como um nome com uma tag opcional (NAME[:TAG]) ou um resumo (NAME@DIGEST). Se omitido em um terminal interativo, o comando solicitará o valor.
--catalog string
O catálogo de destino do Unity Catalog. Se omitido em um terminal interativo, o comando solicitará o valor.
--schema string
O esquema de destino do Unity Catalog. Se omitido em um terminal interativo, o comando solicitará o valor.
--artifact string
O artefato de destino como ARTIFACT[:TAG]. Usa por default o nome e a tag da imagem de origem, ou a tag latest quando a origem não tiver tag.
--pull
Puxe a imagem de origem de seu registro antes de enviar em vez de reutilizar uma imagem local disponível. Default: false.
databricks air convert-to-dabs
Converter uma configuração YAML de carga de trabalho em um Pacote de Ativos do Databricks. Este comando local grava databricks.yml e arquivos de inicialização em generated_artifacts/.
Este comando é uma ponte opcional para um fluxo de trabalho de produção avançado. Você pode continuar a usar databricks air run sem converter a configuração.
Para o fluxo de trabalho do bundle, consulte Programar workloads de GPU e compor tarefas.
databricks air convert-to-dabs YAML_PATH [flags]
Argumentos
YAML_PATH
O caminho para a configuração YAML da carga de trabalho.
Opções
--output-dir string
Directory where the comando writes the bundle. The code source must be inside this directory. Default: the directory that contains the input YAML file.
--force
Sobrescrever os arquivos de bundle gerados que já existem. Default: false.
Sinalizadores globais
--debug
Se o registro de depuração deve ser ativado.
-h ou --help
Display help for the Databricks CLI or the related comando group or the related comando.
--log-file string
Uma strings que representa o arquivo para o qual gravar os Logs de saída. Se esse sinalizador não for especificado, o default será gravar os logs de saída em stderr.
--log-format Formato
The log format type, text or json. O valor default é text.
--log-level string
Uma string que representa o nível do formato de Logs. Se não for especificado, o nível do formato de Logs será desabilitado.
-o, --output Tipo
The comando output type, text or json. The default value is text.
-p, --profile string
O nome do perfil no arquivo ~/.databrickscfg a ser usado para a execução do comando. Se essa flag não for especificada, o perfil denominado DEFAULT será usado, se existir.
--progress-format Formato
O formato para exibir os logs de progresso: default, append, inplace ou json
-t, --target string
If applicable, the bundle target to use
Migrar da CLI do Python legada
A CLI do Databricks e a CLI air baseada em Python preterida usam nomes de comandos e sinalizadores diferentes. Atualize os scripts usando os seguintes mapeamentos:
Python CLI | CLI do Databricks |
|---|---|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|