安裝 AI 執行階段 CLI

Important

AI 執行時的 CLI 目前處於 測試階段

本頁說明如何安裝 air CLI 並設定針對 Databricks 工作區的認證。

Requirements

  • Python 3.10 或更新版本。
  • 一個啟用 AI 執行環境的 Databricks 工作區。 請參閱 需求
  • Databricks CLI,用於管理 ~/.databrickscfg 中的認證設定檔。

安裝 CLI

Databricks 建議用 uv 安裝 CLI:

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

uv tool install 會把 air 放在獨立的環境中,並暴露在你的 PATH,這樣它就不會和你用來訓練程式碼的 Python 直譯器衝突。

--python 3.12 建議但可選。 若未指定Python版本,uv 將使用符合套件Python約束的最新可用版本。

如果你還沒安裝 uv,請先安裝:

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

確認安裝:

air --version
air --help

Authenticate

AI 執行時 CLI 會重用 Databricks 的 CLI 認證設定檔。 登入你的工作區,並在系統提示時為個人資料命名:

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

使用 -p 將設定檔名稱傳遞給任何 air 指令。 例如:

air list runs -p my-workspace

或者,在 shell 中設定 DATABRICKS_CONFIG_PROFILE,將設定檔設為預設:

export DATABRICKS_CONFIG_PROFILE=my-workspace

關於所有認證選項,請參見 Databricks CLI 的認證

下一步