Add GPU support for Linux Azure Container Instances

azurerm/resource_arm_container_group.go

@@ -145,6 +145,28 @@ func resourceArmContainerGroup() *schema.Resource {
 							ForceNew: true,
 						},
 
+						"gpu": {
+							Type:     schema.TypeList,
+							Optional: true,
+							MaxItems: 1,
+							ForceNew: true,
+							Elem: &schema.Resource{
+								Schema: map[string]*schema.Schema{
+									"gpu_count": {
+										Type:     schema.TypeInt,
+										Optional: true,
+										ForceNew: true,
+									},
+
+									"gpu_sku": {
+										Type:     schema.TypeString,
+										Optional: true,
+										ForceNew: true,
+									},
+								},
+							},
+						},
+
 						"port": {
 							Type:         schema.TypeInt,
 							Optional:     true,
@@ -529,6 +551,20 @@ func expandContainerGroupContainers(d *schema.ResourceData) (*[]containerinstanc
 			},
 		}
 
+		if v, ok := data["gpu"]; ok {
+			gpu := v.([]interface{})
+			for _, v := range gpu {
+				gpuCount := v.(map[string]interface{})["gpu_count"]
+				gpuSku := v.(map[string]interface{})["gpu_sku"]
+
+				gpus := containerinstance.GpuResource{
+					Count: utils.Int32(int32(gpuCount.(int))),
+					Sku:   containerinstance.GpuSku(gpuSku.(string)),
+				}
+				container.Resources.Requests.Gpu = &gpus
+			}
+		}
+
 		if v, ok := data["ports"].(*schema.Set); ok && len(v.List()) > 0 {
 			var ports []containerinstance.ContainerPort
 			for _, v := range v.List() {
@@ -779,6 +815,16 @@ func flattenContainerGroupContainers(d *schema.ResourceData, containers *[]conta
 				if v := resourceRequests.MemoryInGB; v != nil {
 					containerConfig["memory"] = *v
 				}
+				if v := resourceRequests.Gpu; v != nil {
+					gpus := make([]interface{}, 0)
+					gpu := make(map[string]interface{})
+
+					gpu["gpu_count"] = *v.Count
+					gpu["gpu_sku"] = v.Sku
+					gpus = append(gpus, gpu)
+
+					containerConfig["gpu"] = gpus
+				}
 			}
 		}
 

azurerm/resource_arm_container_group_test.go

@@ -211,6 +211,9 @@ func TestAccAzureRMContainerGroup_linuxComplete(t *testing.T) {
 					resource.TestCheckResourceAttr(resourceName, "container.0.secure_environment_variables.%", "2"),
 					resource.TestCheckResourceAttr(resourceName, "container.0.secure_environment_variables.secureFoo", "secureBar"),
 					resource.TestCheckResourceAttr(resourceName, "container.0.secure_environment_variables.secureFoo1", "secureBar1"),
+					resource.TestCheckResourceAttr(resourceName, "container.0.gpu.#", "1"),
+					resource.TestCheckResourceAttr(resourceName, "container.0.gpu.0.gpu_count", "1"),
+					resource.TestCheckResourceAttr(resourceName, "container.0.gpu.0.gpu_sku", "K80"),
 					resource.TestCheckResourceAttr(resourceName, "container.0.volume.#", "1"),
 					resource.TestCheckResourceAttr(resourceName, "container.0.volume.0.mount_path", "/aci/logs"),
 					resource.TestCheckResourceAttr(resourceName, "container.0.volume.0.name", "logs"),
@@ -592,7 +595,7 @@ resource "azurerm_container_group" "test" {
     ports {
       port     = 80
       protocol = "TCP"
-    }
+		}
 
     environment_variables = {
       "foo"  = "bar"
@@ -687,7 +690,12 @@ resource "azurerm_container_group" "test" {
     ports {
       port     = 80
       protocol = "TCP"
-    }
+		}
+
+		gpu = {
+			gpu_count = 1
+			gpu_sku = "K80"
+		}
 
     volume {
       name       = "logs"

website/docs/r/container_group.html.markdown

@@ -55,7 +55,7 @@ resource "azurerm_container_group" "aci-helloworld" {
     }
     ports {
       port     = 443
-      protocol = "TCP" 
+      protocol = "TCP"
     }
 
     environment_variables = {
@@ -134,6 +134,10 @@ A `container` block supports:
 
 * `memory` - (Required) The required memory of the containers in GB. Changing this forces a new resource to be created.
 
+~> **Note:** Gpu resources are currently only supported in Linux containers.
+
+* `gpu` - (Optional) Specify to deploy the container with a GPU resource.
+
 * `ports` - (Optional) A set of public ports for the container. Changing this forces a new resource to be created. Set as documented in the `ports` block below.
 
 * `environment_variables` - (Optional) A list of environment variables to be set on the container. Specified as a map of name/value pairs. Changing this forces a new resource to be created.
@@ -184,6 +188,14 @@ A `ports` block supports:
 
 * `protocol` - (Required) The network protocol associated with port. Possible values are `TCP` & `UDP`.
 
+--
+
+A `gpu` block supports:
+
+* `gpu_count` - (Required) The number of GPUs. Allowed values are 1, 2, or 4.
+
+* `gpu_sku` - (Required) The GPU SKU. Allowed values are K80, P100, or V100.
+
 ---
 
 A `volume` block supports: