TensorFlow ジョブで GPU を使用する場合、コンテナビルド用にカスタムベースイメージを指定します。これにより、run 中の正しい GPU 利用が保証されます。リソース設定のDocumentation Index
Fetch the complete documentation index at: https://wb-21fd5541-update-training-api-26.mintlify.app/llms.txt
Use this file to discover all available pages before exploring further.
builder.accelerator.base_image キーの下にイメージタグを追加します。例えば:
base_image の親キーとして accelerator の代わりに cuda を使用してください。