Important
この機能は パブリック プレビュー段階です。
Databricks CLI内のairコマンドグループは、トレーニングや推論などのGPUワークロードをAIランタイムで提出・管理します。
ワークロードの設定フィールドについては、 Workload YAMLリファレンスを参照してください。 最初のワークロードを提出するには、 DatabricksのAIランタイムクイックスタートをご覧ください。
Databricks Air Run
YAML構成からワークロードを提出します。 出力にはジョブ実行IDとジョブ実行URLが含まれます。
databricks air run --file YAML_PATH [flags]
Arguments
None
オプション
-f, --file string
ワークロードのYAML設定へのパス。 このオプションは必須です。
--watch
ストリームノード0は実行が終端状態に達するまでログを続けます。 既定値: false。
--override stringArray
FIELD_PATH=VALUEを使って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
databricks air runが返却したジョブ実行IDです。
オプション
Databricks Air List
実行の一覧を表示します。 デフォルトでは、コマンドは現在のユーザーが提出したアクティブな実行を一覧化します。
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 Air Logs
配信やログの取ってきて、ランをする。
databricks air logs JOB_RUN_ID [flags]
Arguments
JOB_RUN_ID
databricks air runが返却したジョブ実行IDです。
オプション
--node int
ゼロベースのノードインデックスを選択します。 既定値: 0。
--tail int
印刷すべき既存のログラインの数。 既定値: 10000。
--minutes int
直近の数分のログを取り出します。
--tailと組み合わせることはできません。 既定値: 0。
--retry int
再試行を選択してください。
0 は最初の試みであり、 -1 が最新のものである。 既定値: -1。
--download-to string
完全なログをディレクトリにダウンロードしてください。 設定されていない限り、すべてのノードをダウンロード --node 。
--tailまたは--minutesと組み合わせることはできません。
Databricks エアキャンセル
ジョブ実行IDで1回以上の実行をキャンセルするか、現在ユーザーが提出したすべてのアクティブな実行をキャンセルできます。
databricks air cancel [JOB_RUN_ID...] [flags]
Arguments
JOB_RUN_ID
1つ以上のジョブランIDです。
--allと組み合わせることはできません。
オプション
--all
現在のユーザーが提出したすべてのアクティブなランをキャンセルしてください。 既定値: false。
-y, --yes
--allが使った確認プロンプトは飛ばしてください。 既定値: false。
Databricks Air Pools リスト
現在のワークスペースで利用可能なプロビジョニング済みGPU容量をリストアップしてください。
compute.pool_idのプロビジョニング済み容量IDを使ってください。
databricks air pools list [flags]
Arguments
None
オプション
Databricks Air Pools Get
プロビジョニングされたGPU容量の詳細を入手できます。
databricks air pools get [POOL_ID] [flags]
Arguments
POOL_ID
プロビジョニング済み容量ID。 この議論は、ワークスペースにちょうど1つのプロビジョニング容量がある場合にのみ任意です。
オプション
Databricks Air Images Push
コンテナイメージをArtifact Registryにプッシュします。
画像のプッシュに関する前提条件や手順については、 Databricks CLIでプッシュイメージをご参照ください。
databricks air images push [flags]
Arguments
None
オプション
-s, --source string
ソースコンテナイメージは、オプションタグ(NAME[:TAG])またはダイジェスト(NAME@DIGEST)を付けた名前として使います。 インタラクティブ端末で省略すると、コマンドは値のプロンプトを提示します。
--catalog string
目的地はUnityカタログカタログです。 インタラクティブ端末で省略すると、コマンドは値のプロンプトを提示します。
--schema string
宛先のUnity Catalogスキーマ。 インタラクティブ端末で省略すると、コマンドは値のプロンプトを提示します。
--artifact string
目的地のアーティファクトは ARTIFACT[:TAG]。 ソース画像名とタグをデフォルトで設定し、タグがない場合は latest タグを使用します。
--pull
プッシュする前に、利用可能なローカルイメージを再利用するのではなく、レジストリからソースイメージを取得しましょう。 既定値: false。
Databricks Air Convert-to-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
ディレクトリでは、コマンドがバンドルを書き込む。 コードソースはこのディレクトリの中にある必要があります。 デフォルト:入力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、 append、 inplace、または json
-t, --target 文字列
該当する場合は、使用するバンドル ターゲット
レガシーのPython CLIからの移行
DatabricksのCLIと、廃止されたPythonベースのairCLIは異なるコマンド名とフラグを使用しています。 以下のマッピングを使ってスクリプトを更新します:
| Python CLI | Databricks コマンドラインインターフェース (CLI) |
|---|---|
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> |