AI Runtime CLI
備考
ベータ版
AI Runtime CLI は ベータ版です。
air コマンドラインインターフェースは、分散トレーニングワークロードを AI Runtime (オンデマンドのサーバレス GPU コンピュートプラットフォーム) 上で提出および管理を行います。CLI は YAML ベースのジョブ構成を使用し、MLflow と統合し、ワークスペースベースおよび Git ベースのコードワークフローをサポートします。
CLIの使用方法
AI Runtime CLI を使用したい場合:
- ノートブックを開かずに、ラップトップやコードエディタからGPUトレーニングワークロードを提出できます。
- トレーニングジョブをYAMLで宣言的に定義することで、ソース管理に登録できます。
- インタラクティブセッションよりも長く実行する必要がある、または複数のノードにまたがるワークロードである長時間実行のトレーニングジョブやマルチノード分散トレーニングを実行します。
ノートブック内のPython API(@distributedおよび@ray_launch)については、代わりにマルチGPUワークロードを参照してください。
ワークロードを送信するのではなく、ターミナルまたは IDE から SSH 経由で単一 GPU ノードで対話的に作業するには、databricks ssh connect を使用します。SSHトンネルを使用してDatabricksに接続するを参照してください。