Referensi: Uji fungsionalitas GPU

Contoh ketentuan tugas berikut menguji apakah beban kerja GPU AMI yang dideskripsikan dalam Gunakan beban kerja GPU AMI sudah dikonfigurasi dengan benar. Contoh definisi pekerjaan ini menjalankan contoh pengklasifikasi MNIST 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 teks JSON 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