From 5ca781c1e54493672b99197dd345d57004a11a1f Mon Sep 17 00:00:00 2001 From: wangdepeng Date: Tue, 16 Apr 2024 18:46:17 +0800 Subject: [PATCH] Update kubeflow sample pytorchjob to pytorch 2.x and cuda 12.x (#1910) Signed-off-by: wangdepeng --- site/static/examples/jobs/sample-pytorchjob.yaml | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/site/static/examples/jobs/sample-pytorchjob.yaml b/site/static/examples/jobs/sample-pytorchjob.yaml index 37acc3bc7f..bbb5c9e28c 100644 --- a/site/static/examples/jobs/sample-pytorchjob.yaml +++ b/site/static/examples/jobs/sample-pytorchjob.yaml @@ -14,7 +14,9 @@ spec: spec: containers: - name: pytorch - image: docker.io/kubeflowkatib/pytorch-mnist:v1beta1-45c5727 + image: docker.io/kubeflowkatib/pytorch-mnist-cpu:v1beta1-21320b6 +# If you have gpu, pytorch-mnist-gpu would be helpful. pytorch-mnist-gpu is approximately 22GB +# image: docker.io/kubeflowkatib/pytorch-mnist-cpu:latest imagePullPolicy: Always command: - "python3" @@ -31,7 +33,9 @@ spec: spec: containers: - name: pytorch - image: docker.io/kubeflowkatib/pytorch-mnist:v1beta1-45c5727 + image: docker.io/kubeflowkatib/pytorch-mnist-cpu:v1beta1-21320b6 +# If you have gpu, pytorch-mnist-gpu would be helpful. pytorch-mnist-gpu is approximately 22GB +# image: docker.io/kubeflowkatib/pytorch-mnist-cpu:latest imagePullPolicy: Always command: - "python3"