Skip to content

Commit d7c81bd

Browse files
authored
AWS fallback ubuntu (#84)
1 parent 0a564ea commit d7c81bd

File tree

2 files changed

+22
-6
lines changed

2 files changed

+22
-6
lines changed

iterative/aws/provider.go

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,25 @@ func ResourceMachineCreate(ctx context.Context, d *schema.ResourceData, m interf
5656
return decodeAWSError(region, err)
5757
}
5858
if len(imagesRes.Images) == 0 {
59-
return errors.New(ami + " ami not found in region")
59+
imagesRes, err = svc.DescribeImages(&ec2.DescribeImagesInput{
60+
Filters: []*ec2.Filter{
61+
{
62+
Name: aws.String("name"),
63+
Values: []*string{aws.String("*ubuntu/images/hvm-ssd/ubuntu-bionic-18.04*")},
64+
},
65+
{
66+
Name: aws.String("architecture"),
67+
Values: []*string{aws.String("x86_64")},
68+
},
69+
},
70+
})
71+
72+
if err != nil {
73+
return decodeAWSError(region, err)
74+
}
75+
if len(imagesRes.Images) == 0 {
76+
return errors.New("Nor " + ami + " nor Ubuntu Server 18.04 are available in your region")
77+
}
6078
}
6179

6280
sort.Slice(imagesRes.Images, func(i, j int) bool {

iterative/resource_runner.go

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,8 @@ systemctl disable apt-daily-upgrade.service
197197
198198
sudo add-apt-repository universe -y
199199
sudo add-apt-repository ppa:git-core/ppa -y
200-
sudo apt update && sudo apt-get install -y git
200+
sudo apt update && sudo apt install -y software-properties-common git
201+
201202
sudo curl -fsSL https://get.docker.com -o get-docker.sh && sudo sh get-docker.sh
202203
sudo usermod -aG docker ubuntu
203204
sudo setfacl --modify user:ubuntu:rw /var/run/docker.sock
@@ -209,17 +210,14 @@ sudo apt update && sudo apt-get install -y terraform
209210
curl -sL https://deb.nodesource.com/setup_12.x | sudo bash
210211
sudo apt update && sudo apt-get install -y nodejs
211212
212-
sudo apt install -y ubuntu-drivers-common git
213213
sudo ubuntu-drivers autoinstall
214-
215214
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
216215
curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
217216
sudo apt update && sudo apt install -y nvidia-docker2
218-
219217
sudo systemctl restart docker
220-
221218
sudo nvidia-smi
222219
sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
220+
223221
{{end}}
224222
225223
sudo npm install -g git+https://github.com/iterative/cml.git

0 commit comments

Comments
 (0)