air コマンド グループ

備考

プレビュー

この機能は パブリック プレビュー段階です。

Databricks CLI内のairコマンド グループは、AI ランタイムを使用して、トレーニングや推論などの GPU ワークロードを送信および管理します。

注記

これらのコマンドは Google Cloud ではサポートされていません。

databricks air ラン

YAML構成からワークロードを送信します。出力には、ジョブランIDとジョブランURLが含まれます。

databricks air run --file YAML_PATH [flags]

引数

なし

オプション

-f, --file string

ワークロードYAML構成へのパス。このオプションは必須です。

--watch

ランが終了状態に達するまで、ノード0のLogsをストリームします。default: false。

--override stringArray

FIELD_PATH=VALUEを使用してYAMLフィールドを上書きします。複数のフィールドを上書きするには、このオプションを繰り返します。

--dry-run

ワークロードを送信せずに構成を検証します。Default：false。

--idempotency-key string

キーが以前に使用されていた場合に、既存のランを返します。YAML の idempotency_token を上書きします。最大：64文字。

グローバル フラグ

例

次の例では、train.yamlからワークロードを送信します：

Bash

databricks air run --file train.yaml

ワークロード構成のヘルプを表示するには、構成パスを別の引数として -h に渡します。

Bash

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]

引数

JOB_RUN_ID

databricks air runによって返されるジョブのランのID。

オプション

グローバル フラグ

databricks air list

ランの一覧を表示します。defaultでは、このコマンドは現在のユーザーによって送信されたアクティブなランを一覧表示します。

databricks air list [flags]

引数

なし

オプション

--limit int

表示するランの最大数。Default: 20。

--all-status

ターミナル状態のランを含めます。Default：false。

--all-users

他のユーザーによって送信されたランを含めます。Default：false。

--filter stringArray

KEY=VALUEを使用してランをフィルター処理します。複数の条件を要求するには、このオプションを繰り返します。サポートされているキーは、accelerator_type、experiment、num_accelerators、userです。

グローバル フラグ

databricks air logs

ランのLogsをストリームまたは取得します。

databricks air logs JOB_RUN_ID [flags]

引数

JOB_RUN_ID

databricks air runによって返されるジョブのランのID。

オプション

--node int

0ベースのノードインデックスを選択します。Default：0。

--tail int

出力する既存のログ行の数。Default：10000。

--minutes int

過去数分間のLogsを取得します。--tail と組み合わせることはできません。default：0。

--retry int

再試行回数を選択します。0 が初期試行であり、-1 が最新の試行です。Default: -1。

--download-to string

完全な Logs をディレクトリに download します。--nodeが設定されていない限り、すべてのノードを download します。--tailまたは--minutesと組み合わせることはできません。

グローバル フラグ

databricks air cancel

ジョブ実行 ID によって 1 つ以上のランをキャンセルするか、現在のユーザーによって送信されたすべてのアクティブなランをキャンセルします。

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

引数

JOB_RUN_ID

1つ以上のランのID。--allと組み合わせることはできません。

オプション

--all

現在のユーザーによって送信されたすべてのアクティブなランをキャンセルします。Default：false。

-y, --yes

--all で使用される確認プロンプトをスキップします。default：false。

グローバル フラグ

databricks air プール list

現在のワークスペースで使用可能なプロビジョニング済み GPU 容量を一覧表示します。

databricks air pools list [flags]

引数

なし

オプション

グローバル フラグ

databricks air プール get

プロビジョニングされた GPU 容量の詳細を取得します。

databricks air pools get [POOL_ID] [flags]

引数

POOL_ID

プロビジョニング済み容量 ID。この引数が任意となるのは、ワークスペースにプロビジョニング済み容量がちょうど 1 つ存在する場合のみです。

オプション

グローバル フラグ

databricks air images push

コンテナイメージをアーティファクトレジストリにプッシュします。

databricks air images push [flags]

引数

なし

オプション

-s, --source string

オプションのタグ (NAME[:TAG]) またはダイジェスト (NAME@DIGEST) を持つ名前としてのソースコンテナイメージ。対話型ターミナルで省略した場合、コマンドは値の入力を求めます。

--catalog string

宛先の Unity Catalog カタログ。インタラクティブターミナルで省略した場合、コマンドは値の入力を求めます。

--schema string

宛先 Unity Catalog スキーマ。対話型ターミナルで省略した場合、コマンドは値の入力を求めます。

--artifact string

宛先のアーティファクトをARTIFACT[:TAG]として指定します。defaultでは、ソースイメージの名前とタグ、またはソースにタグがない場合はlatestタグになります。

--pull

利用可能なローカルイメージを再利用する代わりに、プッシュする前にレジストリからソースイメージをプルします。Default：false。

グローバル フラグ

databricks air convert-to-dabs

ワークロードのYAML構成をDatabricks Asset Bundleに変換します。このローカルコマンドは、generated_artifacts/の下にdatabricks.ymlおよび起動ファイルを書き込みます。

このコマンドは、高度な本番運用ワークフローへのオプションのブリッジです。構成を変換せずに databricks air run の使用を継続できます。

databricks air convert-to-dabs YAML_PATH [flags]

引数

YAML_PATH

ワークロード YAML 設定へのパス。

オプション

--output-dir string

コマンドがバンドルを書き込むディレクトリ。コードのソースはこのディレクトリ内にある必要があります。default: 入力YAMLファイルを含むディレクトリ。

--force

すでに存在する生成済みバンドルファイルを上書きします。Default：false。

グローバル フラグ

グローバル フラグ

--debug

デバッグ ロギングを有効にするかどうか。

-h または --help

Databricks CLI、関連するコマンドグループ、または関連するコマンドのヘルプを表示します。

--log-file string

出力Logsの書き込み先ファイルを表す文字列。このフラグが指定されていない場合、defaultでは出力Logsがstderrに書き込まれます。

--log-format format

ログフォーマットのタイプ（text または json）。default値は text です。

--log-level string

ログ形式のレベルを表す文字列。指定しない場合、Logs形式のレベルは無効になります。

-o, --output タイプ

コマンドの出力タイプ（text または json）。default値は text です。

-p, --profile string

コマンドのランに使用する ~/.databrickscfg ファイル内のプロファイルの名前。このフラグが指定されていない場合、存在していれば DEFAULT という名前のプロファイルが使用されます。

--progress-format format

進行状況のLogsを表示する形式：default、append、inplace、または json

-t, --target string

必要に応じて、使用するバンドル ターゲット

レガシー Python CLI からの移行

Databricks CLIと非推奨のPythonベースのair CLIでは、異なるコマンド名とフラグが使用されます。次のマッピングを使用してスクリプトを更新します：

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>

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>