GPU を使用する TensorFlow ジョブでは、コンテナーのビルド用にカスタムのベースイメージを指定してください。これにより、実行時に GPU を適切に利用できます。リソース設定の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.
builder.accelerator.base_image キーにイメージタグを追加します。たとえば、次のようにします。
base_image の親キーには accelerator ではなく cuda を使用してください。