Prevent Nvidia DaemonSet from running on CPU instances

Author: ospillinger

cortex.sh

@@ -171,10 +171,11 @@ function install_cortex() {
     -e AWS_ACCESS_KEY_ID=$AWS_ACCESS_KEY_ID \
     -e AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY \
     -e CORTEX_CLUSTER=$CORTEX_CLUSTER \
+    -e CORTEX_REGION=$CORTEX_REGION \
     -e CORTEX_NAMESPACE=$CORTEX_NAMESPACE \
+    -e CORTEX_NODE_TYPE=$CORTEX_NODE_TYPE \
     -e CORTEX_LOG_GROUP=$CORTEX_LOG_GROUP \
     -e CORTEX_BUCKET=$CORTEX_BUCKET \
-    -e CORTEX_REGION=$CORTEX_REGION \
     -e CORTEX_IMAGE_ARGO_CONTROLLER=$CORTEX_IMAGE_ARGO_CONTROLLER \
     -e CORTEX_IMAGE_ARGO_EXECUTOR=$CORTEX_IMAGE_ARGO_EXECUTOR \
     -e CORTEX_IMAGE_FLUENTD=$CORTEX_IMAGE_FLUENTD \

manager/install_cortex.sh

@@ -176,6 +176,10 @@ envsubst < manifests/fluentd.yaml | kubectl apply -f - >/dev/null
 envsubst < manifests/operator.yaml | kubectl apply -f - >/dev/null
 envsubst < manifests/cluster-autoscaler.yaml | kubectl apply -f - >/dev/null
 envsubst < manifests/metrics-server.yaml | kubectl apply -f - >/dev/null
-envsubst < manifests/nvidia.yaml | kubectl apply -f - >/dev/null
+
+instance_prefix=$(echo $CORTEX_NODE_TYPE | cut -c1-2)
+if [ "$instance_prefix" = "p2" ] || [ "$instance_prefix" = "p3" ] || [ "$instance_prefix" = "g3" ]; then
+  envsubst < manifests/nvidia.yaml | kubectl apply -f - >/dev/null
+fi
 
 validate_cortex