Instalar a CLI do ambiente de execução de IA

Importante

A CLI do ambiente de execução de IA está em fase Beta.

Instala a CLI air com uv e autentica-a num espaço de trabalho Databricks utilizando um perfil da Databricks CLI. A CLI requer Python 3.10 ou mais recente.

Requirements

  • Python 3.10 ou mais recente.
  • Um espaço de trabalho Databricks com IA Runtime ativado. Consulte Requisitos.
  • O Databricks CLI, que gere perfis de autenticação em ~/.databrickscfg.

Instalar a CLI

A Databricks recomenda instalar o CLI com UV:

uv tool install --force databricks-air --python 3.12

uv tool install coloca air num ambiente isolado e expõe-o no teu PATH, para não entrar em conflito com o interpretador de Python que usas para o teu código de treino.

--python 3.12 é recomendado, mas opcional. Se não especificar a versão Python, uv usa a versão mais recente disponível que satisfaz a restrição de Python do pacote.

Se ainda não tiver uv, instale-o primeiro:

curl -LsSf https://astral.sh/uv/install.sh | sh

Verifique a instalação:

air --version
air --help

Authenticate

A CLI do runtime de IA reutiliza os perfis de autenticação da CLI da Databricks. Inicie sessão no seu espaço de trabalho e nomeie o perfil quando solicitado:

databricks auth login --host https://<your-workspace>.cloud.databricks.com

Indique o nome do perfil em qualquer comando air com -p. Por exemplo:

air list runs -p my-workspace

Alternativamente, configure DATABRICKS_CONFIG_PROFILE na sua shell para definir um perfil como predefinido:

export DATABRICKS_CONFIG_PROFILE=my-workspace

Para todas as opções de autenticação, consulte Autenticação para a CLI Databricks.

Recursos adicionais

Após a instalação, defina cargas de trabalho num ficheiro de configuração train.yaml com dependências em linha. Começa pelo quickstart, depois usa a referência YAML enquanto constróis a tua configuração: