Referencia: prueba de funcionalidad de GPU

En el siguiente ejemplo, la definición de trabajo prueba si la AMI de carga de trabajo de GPU descrita en Uso de una AMI de carga de trabajo de GPU se ha configurado correctamente. Este ejemplo de definición de trabajo ejecuta el ejemplo del clasificador MNIST TensorFlow profundo de. 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"
}

Puede crear un archivo con el texto JSON anterior denominado tensorflow_mnist_deep.json y, a continuación, registrar una definición de AWS Batch trabajo con el siguiente comando:

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