gpu-8-taskdef.json

{
    "containerDefinitions": [
        {
            "logConfiguration": {
                "logDriver": "awslogs",
                "options": {
                    "awslogs-group": "test-task-tensorflow-gpu",
                    "awslogs-region": "us-east-2",
                    "awslogs-stream-prefix": "tensorflow-gpu"
                }
            },
            "cpu": 1024,
            "resourceRequirements": [
                {
                    "type": "GPU",
                    "value": "8"
                }
            ],
            "environment": [
                {
                    "name": "GPU",
                    "value": "8"
                },
                {
                    "name": "BATCH_SIZE",
                    "value": "128"
                }
            ],
            "image": "brentley/tensorflow-gpu:8904d11",
            "essential": true,
            "name": "tensorflow-gpu"
        }
    ],
    "memory": "131072",
    "family": "tensorflow-8-gpu",
    "cpu": "1024",
    "placementConstraints": []
}