Dockerfile を指定して、W&B に Docker イメージをビルドさせることはできますか？

はい。独自の Dockerfile を指定して、W&B Launch に run 用のイメージをビルドさせることができます。これにより、Launch を使用してジョブをキューに入れて実行しながら、ビルド環境を制御できます。この方法は、requirements は安定している一方でコードベースが変化する project に適しています。

Dockerfile はマウントを使用する形式で記述してください。詳しくは、Docker mounts documentationを参照してください。

Dockerfile を設定したら、次のいずれかの方法で W&B に指定します。

• Dockerfile.wandb を使用する
• W&B CLI を使用する
• W&B App を使用する

Dockerfile.wandb

W&B run のエントリポイントと同じディレクトリに Dockerfile.wandb ファイルを配置します。W&B は組み込みの Dockerfile の代わりにこのファイルを使用します。

W&B CLI

ジョブをキューに入れるには、wandb launch コマンドで --dockerfile フラグを使用します。

wandb launch --dockerfile path/to/Dockerfile

W&B App

W&B App でジョブをキューに追加する際は、Overrides セクションで Dockerfile のパスを指定します。キーを "dockerfile"、値をそのパスとするキーと値のペアとして入力します。

次の JSON は、ローカルディレクトリの Dockerfile を含める方法を示しています。

W&B App の Launch ジョブ

{
  "args": [],
  "run_config": {
    "lr": 0,
    "batch_size": 0,
    "epochs": 0
  },
  "entrypoint": [],
  "dockerfile": "./Dockerfile"
}