W&B Launch で TensorFlow を GPU 上で動作させる方法

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

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

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

最終更新日 2026年9月30日