Task definition to run a GPU enabled task in ECS. Similar behavior to the nvidia-docker project.

gputask.json

{
  "containerDefinitions": [
    {
      "memory": 2048,
      "essential": true,
      "mountPoints": [
        {
          "containerPath": "/hostlib",
          "sourceVolume": "hostlib",
          "readOnly": true
        },
        {
          "containerPath": "/hostusr",
          "sourceVolume": "hostusr",
          "readOnly": true
        }
      ],
      "privileged": true,
      "name": "gputask",
      "image": "your.docker.io/someimage:latest",
      "cpu": 8192
    }
  ],
  "volumes": [
    {
      "host": {
        "sourcePath": "/usr"
      },
      "name": "hostusr"
    },
    {
      "host": {
        "sourcePath": "/lib"
      },
      "name": "hostlib"
    }
  ],
  "family": "gputask"
}