@@ -115,6 +115,7 @@ export CORTEX_NODE_TYPE="${CORTEX_NODE_TYPE:-t3.medium}"
115115export CORTEX_NODES_MIN=" ${CORTEX_NODES_MIN:- 2} "
116116export CORTEX_NODES_MAX=" ${CORTEX_NODES_MAX:- 5} "
117117
118+ export CORTEX_IMAGE_MANAGER=" ${CORTEX_IMAGE_MANAGER:- cortexlabs/ manager: $CORTEX_VERSION_STABLE } "
118119export CORTEX_IMAGE_ARGO_CONTROLLER=" ${CORTEX_IMAGE_ARGO_CONTROLLER:- cortexlabs/ argo-controller: $CORTEX_VERSION_STABLE } "
119120export CORTEX_IMAGE_ARGO_EXECUTOR=" ${CORTEX_IMAGE_ARGO_EXECUTOR:- cortexlabs/ argo-executor: $CORTEX_VERSION_STABLE } "
120121export CORTEX_IMAGE_FLUENTD=" ${CORTEX_IMAGE_FLUENTD:- cortexlabs/ fluentd: $CORTEX_VERSION_STABLE } "
@@ -145,15 +146,15 @@ function install_eks() {
145146 -e CORTEX_NODE_TYPE=$CORTEX_NODE_TYPE \
146147 -e CORTEX_NODES_MIN=$CORTEX_NODES_MIN \
147148 -e CORTEX_NODES_MAX=$CORTEX_NODES_MAX \
148- cortexlabs/manager
149+ $CORTEX_IMAGE_MANAGER
149150}
150151
151152function uninstall_eks() {
152153 docker run --entrypoint /root/uninstall_eks.sh \
153154 -e AWS_ACCESS_KEY_ID=$AWS_ACCESS_KEY_ID \
154155 -e AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY \
155156 -e CORTEX_CLUSTER=$CORTEX_CLUSTER \
156- cortexlabs/manager
157+ $CORTEX_IMAGE_MANAGER
157158}
158159
159160function install_cortex() {
@@ -181,7 +182,7 @@ function install_cortex() {
181182 -e CORTEX_IMAGE_TF_TRAIN_GPU=$CORTEX_IMAGE_TF_TRAIN_GPU \
182183 -e CORTEX_IMAGE_CLUSTER_AUTOSCALER=$CORTEX_IMAGE_CLUSTER_AUTOSCALER \
183184 -e CORTEX_ENABLE_TELEMETRY=$CORTEX_ENABLE_TELEMETRY \
184- cortexlabs/manager
185+ $CORTEX_IMAGE_MANAGER
185186}
186187
187188function uninstall_cortex() {
@@ -190,7 +191,7 @@ function uninstall_cortex() {
190191 -e AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY \
191192 -e CORTEX_CLUSTER=$CORTEX_CLUSTER \
192193 -e CORTEX_NAMESPACE=$CORTEX_NAMESPACE \
193- cortexlabs/manager
194+ $CORTEX_IMAGE_MANAGER
194195}
195196
196197function uninstall_operator() {
@@ -199,7 +200,7 @@ function uninstall_operator() {
199200 -e AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY \
200201 -e CORTEX_CLUSTER=$CORTEX_CLUSTER \
201202 -e CORTEX_NAMESPACE=$CORTEX_NAMESPACE \
202- cortexlabs/manager
203+ $CORTEX_IMAGE_MANAGER
203204}
204205
205206function info() {
@@ -208,7 +209,7 @@ function info() {
208209 -e AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY \
209210 -e CORTEX_CLUSTER=$CORTEX_CLUSTER \
210211 -e CORTEX_NAMESPACE=$CORTEX_NAMESPACE \
211- cortexlabs/manager
212+ $CORTEX_IMAGE_MANAGER
212213}
213214
214215# ###############
0 commit comments