Important
AI ランタイム CLI は ベータ版です。
air を使用して CLI をインストールし、Databricks CLI プロファイルを使用して Databricks ワークスペースに対して認証します。 CLI には、Python 3.10 以降が必要です。
Requirements
- Python 3.10 以降。
- AI ランタイムが有効になっている Databricks ワークスペース。 要件を参照してください。
-
で認証プロファイルを管理する
~/.databrickscfg。
CLI をインストールする
Databricks では、 UV を使用して CLI をインストールすることをお勧めします。
uv tool install --force databricks-air --python 3.12
uv tool install は、独自の分離された環境に air を配置し、それを PATH で公開するため、トレーニング コードに使用するPython インタープリターと競合しません。
--python 3.12 が推奨されますが、省略可能です。 Python バージョンを指定しない場合、uv は、パッケージのPython制約を満たす最新の使用可能なバージョンを使用します。
uvがまだない場合は、最初にインストールします。
curl -LsSf https://astral.sh/uv/install.sh | sh
インストールを確認します。
air --version
air --help
Authenticate
AI ランタイム CLI は、Databricks CLI 認証プロファイルを再利用します。 ワークスペースにログインし、プロンプトが表示されたらプロファイルに名前を付けます。
databricks auth login --host https://<your-workspace>.cloud.databricks.com
airを使用して、任意の-p コマンドにプロファイル名を渡します。 例えば次が挙げられます。
air list runs -p my-workspace
または、シェルで DATABRICKS_CONFIG_PROFILE を設定して、プロファイルを既定にします。
export DATABRICKS_CONFIG_PROFILE=my-workspace
すべての認証オプションについては、 Databricks CLI の認証を参照してください。
その他のリソース
インストール後、インライン依存関係を含む train.yaml 構成でワークロードを定義します。 クイックスタートから始めて、構成をビルドするときに YAML リファレンスを使用します。