Referensi: GPU Fungsionalitas uji

Contoh definisi pekerjaan berikut menguji jika GPU beban kerja yang AMI dijelaskan dalam Gunakan beban GPU kerja AMI dikonfigurasi dengan benar. Contoh definisi pekerjaan ini menjalankan contoh MNIST pengklasifikasi TensorFlow mendalam dari 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"
}

Anda dapat membuat file dengan JSON teks sebelumnya yang disebut tensorflow_mnist_deep.json dan kemudian mendaftarkan definisi AWS Batch pekerjaan dengan perintah berikut:

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