install gpu

Author: DavidGOrtega

cml/local.go

@@ -1,21 +1,34 @@
-//usr/bin/env go run $0 "$@"; exit
 package main
 
 import (
+	"bytes"
 	"fmt"
-
-	localexec "github.com/hashicorp/terraform/builtin/provisioners/local-exec"
-	"github.com/hashicorp/terraform/terraform"
+	"os"
+	"text/template"
 )
 
 func main() {
-	c := terraform.NewResourceConfigRaw(map[string]interface{}{
-		"command": "echo 'Hello world' > hello.txt",
-	})
-	p := localexec.Provisioner()
-	if err := p.Apply(new(terraform.MockUIOutput), nil, c); err != nil {
-		fmt.Println(err)
-	} else {
-		fmt.Println("nice")
-	}
+	data := make(map[string]string)
+	data["AWS_SECRET_ACCESS_KEY"] = os.Getenv("AWS_SECRET_ACCESS_KEY")
+	data["AWS_ACCESS_KEY_ID"] = "slkdksd+sldksldks"
+	data["AZURE_CLIENT_ID"] = "++++++++"
+	data["AZURE_CLIENT_SECRET"] = os.Getenv("AZURE_CLIENT_SECRET")
+	data["AZURE_SUBSCRIPTION_ID"] = os.Getenv("AZURE_SUBSCRIPTION_ID")
+	data["AZURE_TENANT_ID"] = os.Getenv("AZURE_TENANT_ID")
+
+	tmpl, _ := template.New("deploy").Parse(`#!/bin/bash
+	++++
+export AWS_SECRET_ACCESS_KEY={{.AWS_SECRET_ACCESS_KEY}}
+export AWS_ACCESS_KEY_ID={{.AWS_ACCESS_KEY_ID}}
+export AZURE_CLIENT_ID={{.AZURE_CLIENT_ID}}
+export AZURE_CLIENT_SECRET={{.AZURE_CLIENT_SECRET}}
+export AZURE_SUBSCRIPTION_ID={{.AZURE_SUBSCRIPTION_ID}}
+export AZURE_TENANT_ID={{.AZURE_TENANT_ID}}
+sleep 10
+`)
+	var customDataBuffer bytes.Buffer
+	_ = tmpl.Execute(&customDataBuffer, data)
+	fmt.Println(customDataBuffer.String())
+	fmt.Println(os.Getenv("AWS_SECRET_ACCESS_KEY"))
+	fmt.Println(data["AZURE_CLIENT_ID"])
 }

examples/main.tf

@@ -13,6 +13,8 @@ terraform {
 }
 
 provider "iterative" {}
+
+
 resource "iterative_runner" "runner-aws" {
     name = "holy-moly57"
     token = "arszDpb3xtNdKaXmQ6vN"
@@ -25,8 +27,7 @@ resource "iterative_runner" "runner-aws" {
     instance_type = "t2.micro"
 } 
 
-/*
-provider "iterative" {}
+
 resource "iterative_runner" "runner-az" {
     name = "holy-moly56"
     token = "arszDpb3xtNdKaXmQ6vN"
@@ -37,8 +38,7 @@ resource "iterative_runner" "runner-az" {
     cloud = "azure"
     region = "us-west"
     instance_type = "m"
-} 
-*/
+}
 
 
 /*

iterative/resource_runner.go

@@ -6,9 +6,9 @@ import (
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
-	"html/template"
 	"os"
 	"strconv"
+	"text/template"
 
 	"terraform-provider-iterative/iterative/utils"
 
@@ -144,15 +144,6 @@ func resourceRunnerCreate(ctx context.Context, d *schema.ResourceData, m interfa
 	}
 	d.Set("startup_script", startupScript)
 
-	/*
-		diags = append(diags, diag.Diagnostic{
-			Severity: diag.Error,
-			Summary:  d.Get("custom_data").(string),
-		})
-
-		return diags
-	*/
-
 	cloud := d.Get("cloud").(string)
 	if len(cloud) == 0 {
 		diags = append(diags, diag.Diagnostic{
@@ -228,12 +219,31 @@ func provisionerCode(d *schema.ResourceData) (string, error) {
 	data["AZURE_TENANT_ID"] = os.Getenv("AZURE_TENANT_ID")
 
 	tmpl, err := template.New("deploy").Parse(`#!/bin/bash
-export DEBIAN_FRONTEND=noninteractive
+DEBIAN_FRONTEND=noninteractive
+
+{{if .cloud eq "azure"}}
 echo "APT::Get::Assume-Yes \"true\";" | sudo tee -a /etc/apt/apt.conf.d/90assumeyes
-curl -sL https://deb.nodesource.com/setup_12.x | sudo bash
+
+sudo apt update
+sudo curl -fsSL https://get.docker.com -o get-docker.sh && sudo sh get-docker.sh && \
+sudo usermod -aG docker ubuntu
+sudo setfacl --modify user:ubuntu:rw /var/run/docker.sock
+
 curl -fsSL https://apt.releases.hashicorp.com/gpg | sudo apt-key add -
 sudo apt-add-repository "deb [arch=amd64] https://apt.releases.hashicorp.com $(lsb_release -cs) main"
-sudo apt update && sudo apt-get install -y terraform nodejs
+sudo apt update && sudo apt-get install -y terraform
+
+curl -sL https://deb.nodesource.com/setup_12.x | sudo bash
+sudo apt update && sudo apt-get install -y nodejs
+
+sudo apt install -y ubuntu-drivers-common git
+sudo ubuntu-drivers autoinstall 
+sudo rmmod nvidia && sudo nvidia-smi
+curl -s -L https://nvidia.GitHub.io/nvidia-docker/gpgkey | sudo apt-key add - && \
+curl -s -L https://nvidia.GitHub.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
+sudo apt update && sudo apt install -y nvidia-container-toolkit
+{{end}}
+
 sudo npm install -g git+https://github.com/iterative/cml.git#cml-runner
 
 export AWS_SECRET_ACCESS_KEY={{.AWS_SECRET_ACCESS_KEY}}
@@ -242,7 +252,7 @@ export AZURE_CLIENT_ID={{.AZURE_CLIENT_ID}}
 export AZURE_CLIENT_SECRET={{.AZURE_CLIENT_SECRET}}
 export AZURE_SUBSCRIPTION_ID={{.AZURE_SUBSCRIPTION_ID}}
 export AZURE_TENANT_ID={{.AZURE_TENANT_ID}}
-nohup cml-runner{{if .name}} --name {{.name}}{{end}}{{if .labels}} --labels {{.labels}}{{end}}{{if .idle_timeout}} --idle-timeout {{.idle_timeout}}{{end}}{{if .driver}} --driver {{.driver}}{{end}}{{if .repo}} --repo {{.repo}}{{end}}{{if .token}} --token {{.token}}{{end}}{{if .tf_resource}} --tf_resource={{.tf_resource}}{{end}} < /dev/null > std.out 2> std.err &
+nohup cml-runner{{if .name}} --name {{.name}}{{end}}c --labels {{.labels}}{{end}}{{if .idle_timeout}} --idle-timeout {{.idle_timeout}}{{end}}{{if .driver}} --driver {{.driver}}{{end}}{{if .repo}} --repo {{.repo}}{{end}}{{if .token}} --token {{.token}}{{end}}{{if .tf_resource}} --tf_resource={{.tf_resource}}{{end}} < /dev/null > std.out 2> std.err &
 sleep 10
 `)
 	var customDataBuffer bytes.Buffer