

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

# Testar funcionalidade de GPU
<a name="example-test-gpu"></a>

O exemplo de definição de tarefa a seguir testa se a AMI de workload de GPU descrita em [Usar uma AMI de workload de GPU](batch-gpu-ami.md) está configurada corretamente. Este exemplo de definição de trabalho executa o [exemplo](https://github.com/tensorflow/tensorflow/blob/r1.8/tensorflow/examples/tutorials/mnist/mnist_deep.py) do classificador de MNIST profundo do Tensorflow do 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"
}
```

É possível criar um arquivo com o texto JSON anterior chamado `tensorflow_mnist_deep.json` e registrar uma definição de trabalho do AWS Batch com o seguinte comando:

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