參考：測試GPU功能 - AWS Batch

本文為英文版的機器翻譯版本，如內容有任何歧義或不一致之處，概以英文版為準。

參考：測試GPU功能

下列範例任務定義會測試 使用GPU工作負載 AMI 中AMI描述的GPU工作負載是否已正確設定。此範例任務定義會從 執行 TensorFlow 深層MNIST分類器範例 GitHub。

{
    "containerProperties": {
        "image": "tensorflow/tensorflow:1.8.0-devel-gpu",
        "resourceRequirements": [
            {
                "type": "MEMORY",
                "value": "32000"
            },
            {
                "type": "VCPU",
                "value": "8"
            }
        ],
        "command": [
            "sh",
            "-c",
            "cd /tensorflow/tensorflow/examples/tutorials/mnist; python mnist_deep.py"
        ]
    },
    "type": "container",
    "jobDefinitionName": "tensorflow_mnist_deep"
}

您可以建立名為上述JSON文字的檔案，tensorflow_mnist_deep.json然後使用下列命令註冊 AWS Batch 任務定義：

aws batch register-job-definition --cli-input-json file://tensorflow_mnist_deep.json