diff --git a/cortex.sh b/cortex.sh index 1c0b5e3d12..d8502e2132 100755 --- a/cortex.sh +++ b/cortex.sh @@ -192,6 +192,7 @@ function install_cortex() { -e CORTEX_IMAGE_TF_SERVE_GPU=$CORTEX_IMAGE_TF_SERVE_GPU \ -e CORTEX_IMAGE_TF_TRAIN_GPU=$CORTEX_IMAGE_TF_TRAIN_GPU \ -e CORTEX_IMAGE_ONNX_SERVE=$CORTEX_IMAGE_ONNX_SERVE \ + -e CORTEX_IMAGE_ONNX_SERVE_GPU=$CORTEX_IMAGE_ONNX_SERVE_GPU \ -e CORTEX_IMAGE_CLUSTER_AUTOSCALER=$CORTEX_IMAGE_CLUSTER_AUTOSCALER \ -e CORTEX_IMAGE_NVIDIA=$CORTEX_IMAGE_NVIDIA \ -e CORTEX_IMAGE_METRICS_SERVER=$CORTEX_IMAGE_METRICS_SERVER \ diff --git a/manager/install_cortex.sh b/manager/install_cortex.sh index 4397d898d9..6076749a9c 100755 --- a/manager/install_cortex.sh +++ b/manager/install_cortex.sh @@ -64,6 +64,7 @@ function setup_configmap() { --from-literal='IMAGE_TF_TRAIN'=$CORTEX_IMAGE_TF_TRAIN \ --from-literal='IMAGE_TF_SERVE'=$CORTEX_IMAGE_TF_SERVE \ --from-literal='IMAGE_ONNX_SERVE'=$CORTEX_IMAGE_ONNX_SERVE \ + --from-literal='IMAGE_ONNX_SERVE_GPU'=$CORTEX_IMAGE_ONNX_SERVE_GPU \ --from-literal='IMAGE_TF_API'=$CORTEX_IMAGE_TF_API \ --from-literal='IMAGE_PYTHON_PACKAGER'=$CORTEX_IMAGE_PYTHON_PACKAGER \ --from-literal='IMAGE_TF_TRAIN_GPU'=$CORTEX_IMAGE_TF_TRAIN_GPU \