Skip to content

Commit a717f84

Browse files
committed
Update cortex.sh
1 parent 42c056a commit a717f84

File tree

4 files changed

+14
-6
lines changed

4 files changed

+14
-6
lines changed

cortex.sh

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,7 @@ export CORTEX_NODE_TYPE="${CORTEX_NODE_TYPE:-t3.medium}"
115115
export CORTEX_NODES_MIN="${CORTEX_NODES_MIN:-2}"
116116
export CORTEX_NODES_MAX="${CORTEX_NODES_MAX:-5}"
117117

118+
export CORTEX_IMAGE_MANAGER="${CORTEX_IMAGE_MANAGER:-cortexlabs/manager:$CORTEX_VERSION_STABLE}"
118119
export CORTEX_IMAGE_ARGO_CONTROLLER="${CORTEX_IMAGE_ARGO_CONTROLLER:-cortexlabs/argo-controller:$CORTEX_VERSION_STABLE}"
119120
export CORTEX_IMAGE_ARGO_EXECUTOR="${CORTEX_IMAGE_ARGO_EXECUTOR:-cortexlabs/argo-executor:$CORTEX_VERSION_STABLE}"
120121
export 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

151152
function 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

159160
function 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

187188
function 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

196197
function 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

205206
function 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
################

docs/cluster/config.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ export CORTEX_NODES_MAX=5
3636
export CORTEX_NAMESPACE="cortex"
3737

3838
# Image paths
39+
export CORTEX_IMAGE_MANAGER="cortexlabs/manager:master"
3940
export CORTEX_IMAGE_ARGO_CONTROLLER="cortexlabs/argo-controller:master"
4041
export CORTEX_IMAGE_ARGO_EXECUTOR="cortexlabs/argo-executor:master"
4142
export CORTEX_IMAGE_FLUENTD="cortexlabs/fluentd:master"

docs/cluster/install.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
## Prerequisites
44

55
1. [AWS credentials](aws.md)
6+
2. [Docker](https://docs.docker.com/install)
67

78
## Spin up Cortex in your AWS account
89

docs/cluster/uninstall.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,10 @@
11
# Uninstall
22

3+
## Prerequisites
4+
5+
1. [AWS credentials](aws.md)
6+
2. [Docker](https://docs.docker.com/install)
7+
38
## Uninstall Cortex
49

510
<!-- CORTEX_VERSION_MINOR -->

0 commit comments

Comments
 (0)