Guia de introdução rápida da Databricks CLI para o AI Runtime

Importante

Este recurso está no Public Preview.

A CLI Databricks para AI Runtime permite-te submeter, monitorizar e gerir cargas de trabalho de GPU serverless a partir do teu terminal. Este início rápido mostra como configurar e executar uma carga de trabalho do AI Runtime, transmitir registos e inspecionar a execução.

Requirements

Confirme que os comandos air estão disponíveis na CLI Databricks:

databricks air --help

Tip

Para obter ajuda a escrever a configuração da carga de trabalho e a gerir execuções, instala a databricks-ai-runtime competência para o teu agente de IA. Veja como usar um agente de IA.

Execute uma carga de trabalho mínima

Crie um workload.yaml ficheiro que peça uma GPU A10 e execute nvidia-smi para imprimir o nome da GPU:

experiment_name: air-quickstart
compute:
  num_accelerators: 1
  accelerator_type: GPU_1xA10
command: nvidia-smi --query-gpu=name --format=csv,noheader

Submeta a carga de trabalho e transmita os seus logs com databricks air run:

databricks air run --file workload.yaml --watch

A CLI devolve o ID da execução da tarefa e um link para a execução no seu espaço de trabalho. --watch mantém o terminal ligado e transmite os logs até o workload terminar.

Submitting experiment: air-quickstart
Submitted workload with Job Run ID: 783940125630184
View job run at: https://example.cloud.databricks.com/jobs/runs/783940125630184?o=1234567890123456

Monitoring run and streaming logs...
...
Environment setup completed at Thu Oct 1 14:23:45 UTC 2026
NVIDIA A10

Executa o teu próprio código Python

Para carregar e executar código Python local, crie um diretório air-quickstart com os seguintes ficheiros:

air-quickstart/
├── workload.yaml
└── main.py

Tip

Para começar a partir de um projeto executável, em vez de criar estes ficheiros manualmente, clone o AI Runtime cookbook repository e mude para o diretório de início do CLI:

git clone https://github.com/databricks-solutions/ai-runtime-cookbook.git
cd ai-runtime-cookbook/getting-started/air-cli

Utilize o seguinte código Python mínimo para verificar se o AI Runtime executa o código carregado na GPU:

import torch

print("Hello from Python on AI Runtime")
print(f"GPU: {torch.cuda.get_device_name(0)}")

Para estender este exemplo, pode substituir main.py pelo seu próprio treino, inferência em lote ou outra aplicação de IA.

Utilize a seguinte configuração para workload.yaml:

experiment_name: python-quickstart
environment:
  version: 6
  dependencies:
    - torch==2.11.0
compute:
  num_accelerators: 1
  accelerator_type: GPU_1xH100
code_source:
  root_path: .
command: python "$CODE_SOURCE_PATH/main.py"

Esta configuração utiliza as seguintes definições e executa main.py numa única GPU H100:

  • environment.version Selecione a versão 6 do ambiente padrão.
  • environment.dependencies instala o PyTorch 2.11.0 no ambiente selecionado.
  • root_path: . carrega o diretório que contém workload.yaml. Em tempo de execução, $CODE_SOURCE_PATH aponta para o diretório carregado para o servidor.

Submeta a carga de trabalho a partir do diretório que contém workload.yaml:

databricks air run --file workload.yaml --watch

Os registos transmitidos incluem:

...
Hello from Python on AI Runtime
GPU: NVIDIA H100 80GB HBM3

Para executar treino completo, ajustes finos, inferência em lote e cargas de trabalho Ray, clone o AI Runtime Cookbook e execute uma receita.

Para configurar ambientes, fontes de código, computação e outras definições de carga de trabalho, consulte Referência Workload YAML.

Inspecionar uma execução

A CLI imprime o ID da execução da tarefa quando submetes a carga de trabalho. Utilize-o para inspecionar a execução:

databricks air get <job-run-id>

databricks air get Mostra o estado da execução, configuração e tempo. Também está ligado à execução do Job e, quando disponível, à execução do MLflow.

Para transmitir logs em fluxo após o envio ou retomar uma transmissão interrompida, execute:

databricks air logs <job-run-id>

A CLI também fornece comandos para listar e inspecionar execuções, ver registos em tempo real e cancelar execuções ativas. Veja a referência doair comando.

Reference

Recursos adicionais

  • Última atualização a