We need to create a Terraform based ABM cluster setup [1] using GPU on Google Compute Engine. We've made upstream contribution to the Google Terraform Provider for GPU support [2] in the instance_template module release [3]. We need to perform few things --
Provide a config based sample for GPU hardware enabled ABM (anthos-samples repo)
Install NVIDIA CUDA drivers on ABM cluster (anthos-samples repo)
Create a setup guide in this repo on how to create the ABM GPU cluster
We need to create a Terraform based ABM cluster setup [1] using GPU on Google Compute Engine. We've made upstream contribution to the Google Terraform Provider for GPU support [2] in the instance_template module release [3]. We need to perform few things --
[1] https://github.com/GoogleCloudPlatform/anthos-samples/tree/master/anthos-bm-gcp-terraform [2] https://github.com/terraform-google-modules/terraform-google-vm/pull/160 [3] https://github.com/terraform-google-modules/terraform-google-vm/pull/158