Amazon EKS GPU Scale
This repo walks you through the NVIDIA GPU autoscaling with HPA on custom GPU metrics as well as CA(cluster autoscaler) on Amazon EKS.
Prerequisities
Prepare an Amazon EKS cluster with GPU nodes(P2 or P3). You may create the cluster with eksctl or aws-samples/amazon-eks-refarch-cloudformation.
If you prefer to create the cluster with aws-samples/amazon-eks-refarch-cloudformation, create a custom.mk in the local git repository like this
EKS_ADMIN_ROLE ?= arn:aws:iam::903779448426:role/AmazonEKSAdminRole
SSH_KEY_NAME ?= 'aws-pahud'
NodeVolumeSize ?= 30
EnableNodeDrainer ?= no
InstanceTypesOverride ?= 'p2.xlarge,p2.8xlarge,p3.2xlarge'
OnDemandBaseCapacity ?= 0
NodeAutoScalingGroupMinSize ?= 0
NodeAutoScalingGroupDesiredSize ?= 1
NodeAutoScalingGroupMaxSize ?= 10
CLUSTER_STACK_NAME ?= eksdemo-gpu
REGION ?= ap-northeast-1
VPC_ID ?= vpc-e549a281
SUBNET1 ?= subnet-05b643f57a6997deb
SUBNET2 ?= subnet-09e79eb1dec82b7e2
SUBNET3 ?= subnet-0c365d97cbc75ceec
NodeImageId ?= ami-04cf69bbd6c0fae0bExtraNodeLabels=NVIDIAGPU=1 make creat-eks-clusterLabel your GPU nodes
$ kubectl label nodes {NODE_NAME} hardware-type=NVIDIAGPU
# list all GPU nodes. In this sample we only have 1 onde
$ kubectl get no -l hardware-type=NVIDIAGPU
NAME STATUS ROLES AGE VERSION
ip-100-64-71-199.ap-northeast-1.compute.internal Ready <none> 111m v1.13.7-eks-c57ff8Install the Nvidia device plugin
$ kubectl apply -f https://raw.githubusercontent.com/NVIDIA/k8s-device-plugin/1.0.0-beta/nvidia-device-plugin.ymlInstall Helm
follow this guide to install Helm.
Install Prometheus and GPU Node Exporter
Install prometheus-operator, kube-prometheus and GPU metrics dashboard in the default namespace
Identify and label GPU nodes
# Label GPU nodes to run our node-exporter only on GPU nodes.
# Note that a nodeSelector label is defined in node-exporter to control deploying it on GPU nodes only.
kubectl label nodes <gpu-node-name> hardware-type=NVIDIAGPUInstall helm charts
# Install helm https://docs.helm.sh/using_helm/ then run:
helm repo add gpu-helm-charts https://nvidia.github.io/gpu-monitoring-tools/helm-charts
helm repo update
helm install gpu-helm-charts/prometheus-operator --name prometheus-operator
helm install gpu-helm-charts/kube-prometheus --name kube-prometheusGPU metrics Dashboard
# Forward the port for Grafana.
kubectl -n default port-forward $(kubectl get pods -n default -lapp=kube-prometheus-grafana -ojsonpath='{range .items[*]}{.metadata.name}{"\n"}{end}') 3000 &
# Open in browser http://localhost:3000 and go to Nodes DashboardSSH into the GPU node validate the dcgm_* metrics.
Install the Prometheus Adapter to generate custom metrics
$ helm install --name prometheus-adapter --set rbac.create=true,prometheus.url=http://kube-prometheus-prometheus.default.svc.cluster.local,prometheus.port=9090 stable/prometheus-adapterWait a few seconds and you should be able to get custom metrics from the API
$ kubectl get --raw /apis/custom.metrics.k8s.io/v1beta1 | jq -r . | grep dcgm_gpu_utilization
"name": "jobs.batch/dcgm_gpu_utilization",
"name": "services/dcgm_gpu_utilization",
"name": "namespaces/dcgm_gpu_utilization",
"name": "pods/dcgm_gpu_utilization",And check the metrics value like this
$ kubectl get --raw "/apis/custom.metrics.k8s.io/v1beta1/namespaces/default/services/kube-prometheus-exporter-node/dcgm_gpu_utilization" | jq -r .
{
"kind": "MetricValueList",
"apiVersion": "custom.metrics.k8s.io/v1beta1",
"metadata": {
"selfLink": "/apis/custom.metrics.k8s.io/v1beta1/namespaces/default/services/kube-prometheus-exporter-node/dcgm_gpu_utilization"
},
"items": [
{
"describedObject": {
"kind": "Service",
"namespace": "default",
"name": "kube-prometheus-exporter-node",
"apiVersion": "/v1"
},
"metricName": "dcgm_gpu_utilization",
"timestamp": "2019-07-07T04:57:36Z",
"value": "0"
}
]
}deploy the GPU stress testing application
$ kubectl apply -f gputest.yaml
deployment.extensions/gputest configured
# view the deployment
$ kubectl get deploy/gputest
NAME READY UP-TO-DATE AVAILABLE AGE
gputest 1/1 1 1 1m
# list the pods
$ kubectl get po -l run=gputest
NAME READY STATUS RESTARTS AGE
gputest-79988456cb-hn4h6 1/1 Running 0 1mWe should only have a single pod running
Manual scale out the deployment to 2
$ kubectl scale --replicas=2 deploy/gputest
deployment.extensions/gputest scaled
pahud:~/environment/k8s-gpu-hpa $ kubectl get po -l run=gputest
NAME READY STATUS RESTARTS AGE
gputest-79988456cb-7xfpb 0/1 Pending 0 4s
gputest-79988456cb-hn4h6 1/1 Running 0 9m11sThe 2nd pod can't be deployed because each pod will request 1 single GPU and we only have 1 node in the cluster.
Events:
Type Reason Age From Message
---- ------ ---- ---- -------
Warning FailedScheduling 43s (x2 over 43s) default-scheduler 0/1 nodes are available: 1 Insufficient nvidia.com/gpu.
pahud:~/environment/k8s-gpu-hpa $ Let's scale it back to 1 at this moment.
$ kubectl scale --replicas=1 deploy/gputestDeploy the HPA object
Now let's deploy a HPA object to scale out the pods based on the dcgm_gpu_utilization custom metric.
$ kubectl apply -f hpa.yaml
horizontalpodautoscaler.autoscaling/hpa-gpu createdGet the HPA status. The custom metric goes to 100 now and HPA scales out to 2 as the desired replicas of this deployment.
# get the HPA
$ kubectl get hpa/hpa-gpu
NAME REFERENCE TARGETS MINPODS MAXPODS REPLICAS AGE
hpa-gpu Deployment/gputest 100/80 1 3 2 62sdescribe the HPA
$ kubectl describe hpa/hpa-gpu
Name: hpa-gpu
Namespace: default
Labels: <none>
Annotations: kubectl.kubernetes.io/last-applied-configuration:
{"apiVersion":"autoscaling/v2beta1","kind":"HorizontalPodAutoscaler","metadata":{"annotations":{},"name":"hpa-gpu","namespace":"default"},...
CreationTimestamp: Sun, 07 Jul 2019 05:22:56 +0000
Reference: Deployment/gputest
Metrics: ( current / target )
"dcgm_gpu_utilization" on Service/kube-prometheus-exporter-node: 100 / 80
Min replicas: 1
Max replicas: 3
Deployment pods: 2 current / 2 desired
Conditions:
Type Status Reason Message
---- ------ ------ -------
AbleToScale True ReadyForNewScale recommended size matches current size
ScalingActive True ValidMetricFound the HPA was able to successfully calculate a replica count from Service metric dcgm_gpu_utilization
ScalingLimited False DesiredWithinRange the desired count is within the acceptable range
Events:
Type Reason Age From Message
---- ------ ---- ---- -------
Normal SuccessfulRescale 2m33s horizontal-pod-autoscaler New size: 2; reason: Service metric dcgm_gpu_utilization above targetThe HPA is trying to scale out to 2 but the 2nd replica can't be deployed because of insufficent GPU resources.
Cluster Autoscaler
TBD