air 命令群組

air Databricks CLI 中的指令群組負責提交並管理 GPU 工作負載,如訓練與推論,並搭配 AI 執行時。

關於工作負載設定欄位,請參見 工作負載 YAML 參考資料。 要提交你的第一個工作負載,請參閱 AI 執行環境快速啟動。

Databricks 空中跑

從 YAML 設定提交工作負載。 輸出包含工作執行 ID 和工作執行 URL。

databricks air run --file YAML_PATH [flags]

Arguments

None

選項

-f, --file string

    工作負載 YAML 設定的路徑。 這是必要選項。

--watch

    串流節點 0 會記錄直到執行達到終端狀態。 預設值:false。

--override stringArray

    用 FIELD_PATH=VALUES 覆蓋 YAML 欄位。 重複這個選項以覆蓋多個欄位。

--dry-run

    在不提交工作負載的情況下驗證設定。 預設值:false。

--idempotency-key string

    回傳先前使用過的舊行程。 在 YAML 中被 idempotency_token 覆蓋。 最多:64 字元。

全域旗標

Examples

以下範例提交來自 train.yaml的工作負載:

databricks air run --file train.yaml

將設定路徑作為獨立參數 -h 傳遞以顯示工作負載配置說明:

databricks air run -h config
databricks air run -h config.compute
databricks air run -h config.compute.accelerator_type

Databricks air get

取得跑步的細節。

databricks air get JOB_RUN_ID [flags]

Arguments

JOB_RUN_ID

    工作執行 ID 由 databricks air run返回。

選項

全域旗標

Databricks 空中列表

清單執行。 預設情況下,指令會列出目前使用者提交的活躍執行。

databricks air list [flags]

Arguments

None

選項

--limit int

    最多可展示的得分數。 預設值:20。

--all-status

    包含終端狀態下的跑動。 預設值:false。

--all-users

    包含其他用戶提交的作品。 預設值:false。

--filter stringArray

    過濾器執行時使用 KEY=VALUE。 重複這個選項,要求多個條件。 支援的鍵包括 accelerator_type、 experiment、 num_accelerators、 user和 。

全域旗標

Databricks 空中日誌

串流或取木頭來執行。

databricks air logs JOB_RUN_ID [flags]

Arguments

JOB_RUN_ID

    工作執行 ID 由 databricks air run返回。

選項

--node int

    選擇一個以零為基礎的節點索引。 預設值:0。

--tail int

    現有的原稿要列印數量。 預設值:10000。

--minutes int

    取出過去幾分鐘的紀錄。 不能與 --tail結合使用。 預設值:0。

--retry int

    選擇重試嘗試。 0 是最初的嘗試,也是 -1 最新的。 預設值:-1。

--download-to string

    下載完整日誌到目錄。 除非 --node 設定,否則會下載所有節點。 無法與 --tail 或 --minutes結合。

全域旗標

Databricks 空中取消

依據工作執行 ID 取消一個或多個執行,或取消目前使用者提交的所有活躍執行。

databricks air cancel [JOB_RUN_ID...] [flags]

Arguments

JOB_RUN_ID

    一個或多個工作執行的ID。 不能與 --all結合使用。

選項

--all

    取消目前用戶提交的所有活躍跑道。 預設值:false。

-y, --yes

    跳過 。--all 預設值:false。

全域旗標

Databricks 空中池列表

列出目前工作區可用的已配置 GPU 容量。

請使用配置好的容量 ID 。compute.pool_id

databricks air pools list [flags]

Arguments

None

選項

全域旗標

Databricks Air Pools 取得

取得配置 GPU 容量的詳細資訊。

databricks air pools get [POOL_ID] [flags]

Arguments

POOL_ID

    配置容量 ID。 這個參數只有在工作區恰好有一個配置容量時才是可選的。

選項

全域旗標

Databricks Air Images 推送

將容器映像推送到 Artifact Registry。

關於推送映像的前置條件與說明,請參見 Push image with the Databricks CLI。

databricks air images push [flags]

Arguments

None

選項

-s, --source string

    來源容器影像以名稱形式呈現,並可選標籤()NAME[:TAG]或摘要NAME@DIGEST()。 若在互動終端機中省略,指令會提示該值。

--catalog string

    目的地 Unity 目錄。 若在互動終端機中省略,指令會提示該值。

--schema string

    目的地 Unity 目錄架構。 若在互動終端機中省略,指令會提示該值。

--artifact string

    目的文物為 ARTIFACT[:TAG]。 預設為來源圖片名稱和標籤,或當來源沒有標籤時則使用標籤 latest 。

--pull

    在推送前,從其登錄檔拉取原始映像檔,而不是重複使用可用的本地映像檔。 預設值:false。

全域旗標

Databricks Air 轉換成 DAB

將工作負載的 YAML 配置轉換成 Databricks 資產套件。 此本地指令可databricks.yml依 寫入並啟動檔案。generated_artifacts/

此指令是通往進階生產工作流程的可選橋樑。 你可以繼續使用 databricks air run 而不需轉換設定。

關於套件工作流程,請參見 「排程 GPU 工作負載與組合任務」。

databricks air convert-to-dabs YAML_PATH [flags]

Arguments

YAML_PATH

    工作負載 YAML 配置的路徑。

選項

--output-dir string

    指令寫入 bundle 的目錄。 程式碼來源必須在這個目錄中。 預設值:包含輸入 YAML 檔案的目錄。

--force

    覆蓋已存在的組合檔案。 預設值:false。

全域旗標

全域旗標

--debug

  是否要啟用偵錯記錄。

-h 或 --help

    顯示 Databricks CLI、相關命令群組或相關命令的幫助說明。

--log-file 字串

    字串,表示要寫入輸出記錄檔的檔案。 若未指定此旗標,則預設值是將輸出記錄寫入 stderr。

--log-format 格式

    記錄格式類型,text 或 json。 預設值為 text。

--log-level 字串

    表示日誌格式層級的字串。 若未指定,則日誌的格式層級功能將被停用。

-o, --output 類型

    指令輸出類型,text 或 json。 預設值為 text。

-p, --profile 字串

    用於執行命令的~/.databrickscfg檔案中的配置檔名稱。 如果未指定此旗標,則如果存在,則會使用名為 DEFAULT 的配置檔。

--progress-format 格式

    顯示進度記錄的格式:default、、appendinplace、 或json

-t, --target 字串

    如果適用,要使用的套件組合目標

從舊有的 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>

其他資源