使用 Databricks CLI 搭配 AI 執行環境

Important

Databricks 的 AI 執行時 CLI 指令目前已公開 預覽。 工作區管理員必須先啟用 AI 執行時預覽,才能提交工作負載。 請參閱 需求說明。

databricks air指令組負責提交並管理 AI Runtime 上的分散式訓練工作負載,AI Runtime 是按需的無伺服器 GPU 運算平台。 它採用基於 YAML 的工作設定,整合 MLflow,並支援基於工作空間與 git 的程式碼工作流程。

Data bricks 的 CLI 指令用於 AI 執行環境,符合與 AI 執行環境工作負載相同的合規標準。 關於支援的標準與例外,請參見 合規安全設定檔。

從 Python CLI 遷移

Databricks CLI 取代了以databricks-air套件形式發佈的基於 Python air 的 CLI。 Python CLI 沒有新增功能。 安裝最新的 Databricks CLI 以使用最新的 AI 執行時功能。 Databricks CLI 支援 macOS、Linux 及 Windows。

有些指令名稱和旗標有所不同,因此遷移作業不僅僅是加入 databricks 前綴。 請參見 指揮差異。

何時使用 CLI

當您想要將 Databricks CLI 與 AI Runtime 搭配使用時:

  • 不用打開筆記本,也可以從筆電和程式碼編輯器提交 GPU 訓練工作負載。
  • 以宣告式方式在 YAML 中定義訓練作業,以便將其納入版本控制。
  • 執行長時間執行的訓練作業或多節點分散式訓練——這些工作負載需要在互動式工作階段結束後仍持續執行,或橫跨多個節點。

關於筆記本內的 Python API(@distributed和@ray_launch),請參見「分散式筆記本訓練」。

若要透過 SSH 在單一 GPU 節點上互動式工作——從終端機或 IDE 進行,而非提交工作負載——使用 databricks ssh connect。 請參閱 使用 SSH 通道連線至 Databricks。

開始

安裝最新的 Databricks CLI 並設定認證。 請參閱 安裝或更新 Databricks CLI 和 Databricks CLI 的驗證。

然後依照快速入門中的步驟提交你的第一個工作負載。 關於指令與旗標,請參見 air 指令群組。