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> |