Référence : GPU Fonctionnalité de test

L'exemple de définition de tâche suivant teste si la GPU charge de travail AMI décrite dans Utiliser une GPU charge de travail AMI est correctement configurée. Cet exemple de définition de tâche exécute l'exemple de MNIST classificateur TensorFlow profond 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"
}

Vous pouvez créer un fichier avec le JSON texte précédent appelé, tensorflow_mnist_deep.json puis enregistrer une définition de AWS Batch tâche à l'aide de la commande suivante :

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