air grupo de comando
Visualização
Esse recurso está em Prévia Pública.
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.
Estes comandos não são compatíveis com o Google Cloud.
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 pools list
Liste a capacidade de GPU provisionada disponível para o Workspace atual.
databricks air pools list [flags]
Argumentos
Nenhuma
Opções
databricks air pools get
Obter detalhes para a capacidade de GPU provisionada.
databricks air pools get [POOL_ID] [flags]
Argumentos
POOL_ID
O ID de capacidade provisionada. O argumento é opcional apenas quando o workspace tem exatamente uma capacidade provisionada.
Opções
databricks air images push
Enviar uma imagem de contêiner para o Artifact Registry.
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.
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 |
|---|---|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|