Bottlerocket nvidia (aws#1331)

* Add bottlerocket lookup for nvidia and arm64 AMIs

* Add docs update for bottlerocket amifamily

* Separate GPU and arch checks for bottlerocket

* Remove AWSNeurons check for BR AMI

pkg/cloudprovider/aws/ami.go

@@ -108,10 +108,14 @@ func (p *AMIProvider) getAL2Alias(version string, instanceType cloudprovider.Ins
 // getBottlerocketAlias returns a properly-formatted alias for a Bottlerocket AMI from SSM
 func (p *AMIProvider) getBottlerocketAlias(version string, instanceType cloudprovider.InstanceType) string {
 	arch := "x86_64"
+	amiSuffix := ""
+	if !instanceType.NvidiaGPUs().IsZero() {
+		amiSuffix = "-nvidia"
+	}
 	if instanceType.Architecture() == v1alpha5.ArchitectureArm64 {
 		arch = instanceType.Architecture()
 	}
-	return fmt.Sprintf("/aws/service/bottlerocket/aws-k8s-%s/%s/latest/image_id", version, arch)
+	return fmt.Sprintf("/aws/service/bottlerocket/aws-k8s-%s%s/%s/latest/image_id", version, amiSuffix, arch)
 }
 
 // getUbuntuAlias returns a properly-formatted alias for an Ubuntu AMI from SSM

website/content/en/preview/AWS/provisioning.md

@@ -154,7 +154,7 @@ spec:
 
 The AMI used when provisioning nodes can be controlled by the `amiFamily` field. Based on the value set for `amiFamily`, Karpenter will automatically query for the appropriate [EKS optimized AMI](https://docs.aws.amazon.com/eks/latest/userguide/eks-optimized-amis.html) via AWS Systems Manager (SSM). 
 
-Currently, Karpenter supports `amiFamily` values `al2`, `bottlerocket`, and `ubuntu`. GPUs are only supported with `al2`.
+Currently, Karpenter supports `amiFamily` values `al2`, `bottlerocket`, and `ubuntu`. GPUs are only supported with `al2` and `bottlerocket`.
 
 Note: If a custom launch template is specified, then the AMI value in the launch template is used rather than the `amiFamily` value.