設定テンプレートを設定する
設定テンプレートを使用すると、管理者は計算リソースの消費に対するガードレールを管理できます。メモリ使用量、GPU、実行時間などの項目について、デフォルト値、最小値、最大値を設定します。 設定テンプレートでキューを設定すると、チームメンバーは、あなたが定義した項目を、指定した範囲内でのみ変更できます。キューテンプレートを設定する
既存のキューまたは新しいキューで、キューテンプレートを設定できます。次の手順では、キューの設定にテンプレートフィールドを追加し、チームメンバーが定義した制限の範囲内でのみ値を設定できるようにします。- W&B Launch App にアクセスします。
- テンプレートを追加するキューの名の横にある View queue を選択します。
- Config タブを選択します。ここには、キューの作成日時、キューの設定、既存の起動時オーバーライドなど、キューに関する情報が表示されます。
- Queue config セクションにアクセスします。
- テンプレートを作成したい設定のキーと値を特定します。
- 設定内の値をテンプレートフィールドに置き換えます。テンプレートフィールドは
{{variable-name}}の形式です。 - Parse configuration ボタンをクリックします。設定を解析すると、W&B は作成した各テンプレートに対応するタイルを キュー設定 の下に自動的に作成します。
- 生成された各タイルについて、まず キュー設定 で許可するデータタイプ (string、integer、または float) を指定する必要があります。これを行うには、Type ドロップダウンメニューからデータタイプを選択します。
- データタイプに応じて、各タイル内に表示されるフィールドに入力します。
- Save config をクリックします。
launch config
InstanceType のテンプレートフィールドを追加すると、設定は次のようになります。
launch config
aws-instance というラベルの新しいタイルが表示されます。
次に、Type ドロップダウンからデータ型として String を選択します。すると、ユーザーが選択できる値を指定するためのフィールドが表示されます。たとえば、次の画像では、Team Admin がユーザー向けに 2 つの AWS インスタンスタイプ (ml.m4.xlarge と ml.p3.xlarge) を設定しています。

Launch ジョブを動的に設定する
キュー設定は、agent がキューからジョブを取り出す際に評価されるマクロを使って動的に設定できます。マクロを使用すると、起動時に run 固有の値をキュー設定に挿入できます。設定できるマクロは次のとおりです。前の表に記載されていないカスタムマクロ (たとえば
${MY_ENV_VAR}) は、agent の環境変数の値に置き換えられます。アクセラレータまたは GPU 上で実行するイメージを build するために Launch agent を使用する
アクセラレータ環境で実行するイメージを build するために Launch を使用する場合、生成されるイメージを対象のハードウェアと互換性のあるものにするために、アクセラレータのベースイメージを指定する必要があることがあります。 アクセラレータのベースイメージは、次の要件を満たす必要があります。- Debian との互換性。Launch の Dockerfile では、
apt-getを使用してpythonを取得します。 - CPU および GPU のハードウェア命令セットとの互換性。使用する予定の GPU で CUDA のバージョンがサポートされていることを確認してください。
- 指定するアクセラレータのバージョンと、ML アルゴリズムにインストールされているパッケージとの互換性。
- インストールされたパッケージをハードウェアと互換性がある状態にするために必要な追加の設定手順。
TensorFlow で GPU を使用する
TensorFlow が GPU を使用するようにするには、キューのリソース設定でbuilder.accelerator.base_image キーに Docker image とその image tag を指定します。
たとえば、tensorflow/tensorflow:latest-gpu をベースイメージに指定すると、TensorFlow で GPU を正しく使用できます。これは、キューのリソース設定で構成できます。
次の JSON スニペットは、キュー設定で TensorFlow のベースイメージを指定する方法を示しています。
Queue config