Support setting the init-container-image for pytorch-operator (#1097)

Signed-off-by: Syulin7 <[email protected]>
kubeflow · Jun 18, 2024 · 19c9090 · 19c9090
1 parent 48eed0f
commit 19c9090
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/arena-artifacts/charts/pytorch-operator/templates/operator-dp.yaml b/arena-artifacts/charts/pytorch-operator/templates/operator-dp.yaml
@@ -32,6 +32,7 @@ spec:
             - -v=1
             - --monitoring-port=8443
             - --threadiness=4
+            - --init-container-image={{ .Values.initContainerImage }}
           # image: gcr.io/kubeflow-images-public/pytorch-operator:v0.6.0-18-g5e36a57
           image: {{ include "arena.imagePrefix" . }}/{{ .Values.image }}:{{ .Values.tag }}
           imagePullPolicy: {{ .Values.imagePullPolicy }}

diff --git a/arena-artifacts/values.yaml b/arena-artifacts/values.yaml
@@ -76,7 +76,7 @@ mpi:
 pytorch:
   enabled: true
   image: acs/pytorch-operator
-  tag: v1.0-aliyun-5ab19b0
+  tag: v1.0-aliyun-4026101
   imagePullPolicy: IfNotPresent
   resources:
     limits:
@@ -86,6 +86,7 @@ pytorch:
       cpu: 100m
       memory: 256Mi
   nodeSelector: {}
+  initContainerImage: alpine:3.10
 
 # et-operator
 et: