メインコンテンツへスキップ

Documentation Index

Fetch the complete documentation index at: https://wb-21fd5541-john-wbdocs-2044-rename-serverless-products.mintlify.app/llms.txt

Use this file to discover all available pages before exploring further.

GPU を使用する 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 を使用してください。