air 命令群組

Important

這項功能目前處於 公開預覽版。

air Databricks CLI 中的指令群組負責提交並管理 AI 執行時的訓練工作負載。 關於需求與使用指引,請參閱「 使用 Databricks CLI 搭配 AI 執行環境」。 要提交你的第一個工作負載,請參閱 Databricks CLI 的 AI 執行時快速入門。

欲獲得完整的指令與設定參考,請使用 CLI 說明:

databricks air --help                  # All commands
databricks air <command> --help        # Flags for a specific command
databricks air run -h config           # YAML config reference
databricks air run -h config.compute   # Per-field help for a YAML section

CLI 說明書會反映你安裝的版本。 配置說明是從驗證結構中產生的,因此請利用它來檢查欄位類型、需求和約束。

Commands

命令 Purpose
databricks air run 從 YAML 設定提交訓練工作負載。支援 -f (--file)、 --watch、 --dry-run可 --override 重複標誌及 --idempotency-key。
databricks air get <run-id> 顯示指定運行的狀態、配置及時間細節。
databricks air list 列出目前使用者的活躍運行。 可包含 --all-status 已完成的執行次數、 --all-users 其他使用者,以及 --limit 設定結果計數(預設 20)。 重複此 --filter KEY=VALUE 步驟以篩選 accelerator_type、 experiment、 num_accelerators或 user。
databricks air logs <run-id> 從正在進行的執行中串流日誌,或從已完成的執行中擷取日誌。 請使用 --node 來選擇節點、 --retry 選擇嘗試,以及 --download-to 下載完整日誌。
databricks air cancel <run-id> [<run-id> ...] 取消一次或多次跑動。 用 --all 來取消你的主動任務。 --all 除非你通過 --yes,否則會有確認提示。
databricks air convert-to-dabs <yaml-path> 將工作負載 YAML 轉為本地宣告式自動化套件,無需聯絡工作區。 用 --output-dir 來選擇 bundle 目錄並 --force 覆寫已產生的檔案。
databricks air images push 一個輔助工具,用來設定 Docker 認證、標記映像檔,並推送到 Artifact 登錄檔。 版本為 v1.19.0 及更新版本。 請參閱 使用 Databricks CLI 的推送映像檔。

databricks air run -f train.yaml --dry-run 驗證 YAML 架構,並在本地覆蓋,無需提交工作負載或驗證。 它不會檢查本地程式碼路徑或 Git 版本是否存在,也不會驗證工作空間權限或 GPU 可用性。

對於日誌,可以用 --tail 來選擇行數或 --minutes 時間視窗。 這些旗標互斥,無法與 合併。--download-to

全域旗標

Flag Purpose
-p、--profile 使用已命名的 Databricks CLI 認證設定檔,而非預設設定檔。
--debug 啟用除錯日誌。
-o、--output 選擇輸出格式: text 或 json。 預設值為 text。
-t、--target 如適用,選擇該組合目標。

使用 -h 或 --help 在指令中表示幫助。 要顯示 YAML 欄位幫助,請使用 databricks air run -h config.<section>。

與舊有 Python CLI 的差異

Databricks 的 CLI 與舊有的基於 air Python 的 CLI 使用不同的指令名稱與旗標。 使用以下映射更新腳本:

Python CLI Databricks 命令行介面
air run -f train.yaml databricks air run -f train.yaml
air run -f train.yaml --override A=X B=Y databricks air run -f train.yaml --override A=X --override B=Y
air get run <run-id> databricks air get <run-id>
air list runs databricks air list --all-status
air list runs --active databricks air list
air -h config databricks air run -h config
air logs <run-id> databricks air logs <run-id>
air cancel <run-id> databricks air cancel <run-id>

其他資源