GPU 上で TensorFlow と連携して W&B Launch を動作させる方法

GPU で W&B Launch を使用して TensorFlow ジョブを実行するには、コンテナーのビルド用にカスタムのベースイメージを指定してください。これにより、生成されるコンテナーが GPU にアクセスできるようになります。リソース設定の builder.accelerator.base_image キーにイメージタグを追加します。たとえば、次のようにします。

{
    "gpus": "all",
    "builder": {
        "accelerator": {
            "base_image": "tensorflow/tensorflow:latest-gpu"
        }
    }
}

W&B 0.15.6 より前のバージョンでは、base_image の親キーには accelerator ではなく cuda を使用してください。