testing/util/gpu_vm.wf.json

{
  "Name": "gpu-vm",
  "Vars": {
    "test_cfg": { "Required": true, "Description": "Cloud-config to use for the test" }
  },
  "Sources": {
    "cloud-config": "${test_cfg}"
  },
  "Steps": {
    "create-vm": {
      "CreateInstances": [
        {
          "Name": "preload-test",
          "Disks": [{"Source": "boot-disk"}],
          "Metadata": {
            "user-data": "${SOURCE:cloud-config}",
            "block-project-ssh-keys": "TRUE"
          },
          "guestAccelerators": [
            {
              "acceleratorCount": 1,
              "acceleratorType": "https://www.googleapis.com/compute/v1/projects/${PROJECT}/zones/${ZONE}/acceleratorTypes/nvidia-tesla-k80"
            }
          ],
          "scheduling": {
            "onHostMaintenance": "TERMINATE"
          }
        }
      ]
    }
  }
}