{
  "variables": {
    "aws_access_key": "{{env `AWS_ACCESS_KEY_ID`}}",
    "aws_secret_key": "{{env `AWS_SECRET_ACCESS_KEY`}}",
    "region": "eu-west-1",
    "ssh_username": "ubuntu",
    "base_ami": "ami-0bc691261a82b32bc",
    "instance_type": "c5.large",
    "subnet_id": "subnet-c39cb6a5",
    "docker_version": "",
    "cuda_drivers_version": "",
    "build_prefix": "",
    "ami_regions": "",
    "ami_groups": "",
    "image_version": ""
  },
  "builders": [
    {
      "type": "amazon-ebs",
      "access_key": "{{user `aws_access_key`}}",
      "secret_key": "{{user `aws_secret_key`}}",
      "region": "{{user `region`}}",
      "subnet_id": "{{user `subnet_id`}}",
      "source_ami": "{{user `base_ami`}}",
      "instance_type": "{{user `instance_type`}}",
      "ssh_username": "{{user `ssh_username`}}",
      "ami_name": "{{user `build_prefix`}}dstack-cuda-{{user `image_version` | clean_resource_name}}",
      "associate_public_ip_address": true,
      "ami_regions": "{{user `ami_regions`}}",
      "ami_groups": "{{user `ami_groups`}}",
      "tags": {
        "Name": "DSTACK-CUDA"
      },
      "launch_block_device_mappings": [
        {
          "device_name":           "/dev/sda1",
          "volume_size":           "100",
          "volume_type":           "gp2",
          "delete_on_termination": true
        }
      ]
    }
  ],
  "provisioners": [
    {
      "type": "shell",
      "inline": ["cloud-init status --long --wait"]
    },
    {
      "type": "shell",
      "scripts": [
        "provisioners/kernel/apt-upgrade.sh",
        "provisioners/kernel/apt-daily.sh",
        "provisioners/kernel/apt-packages.sh",
        "provisioners/kernel/kernel-tuning.sh"
      ]
    },
    {
      "type": "file",
      "source": "provisioners/install-docker.sh",
      "destination": "/tmp/install-docker.sh"
    },
    {
      "type": "file",
      "source": "provisioners/run-docker",
      "destination": "/tmp/run-docker"
    },
    {
      "type": "shell",
      "inline": [
        "cd /tmp",
        "chmod +x install-docker.sh",
        "./install-docker.sh --version {{user `docker_version`}}"]
    },
    {
      "type": "shell",
      "environment_vars": ["CUDA_DRIVERS_VERSION={{user `cuda_drivers_version`}}"],
      "script": "provisioners/cuda.sh"
    },
    {
      "type": "shell",
      "script": "provisioners/install-nvidia-container-toolkit.sh"
    }
  ]
}