メインコンテンツへスキップ
GPU を使用する TensorFlow ジョブの場合、コンテナビルドにカスタムベースイメージを指定してください。これにより、 Runs の実行中に GPU が適切に活用されるようになります。リソース 設定 の builder.accelerator.base_image キー の下にイメージタグを追加します。例:
{
    "gpus": "all",
    "builder": {
        "accelerator": {
            "base_image": "tensorflow/tensorflow:latest-gpu"
        }
    }
}
W&B の バージョン 0.15.6 より前の場合は、base_image の親 キー として accelerator の代わりに cuda を使用してください。