publish torchrun example via Dockerfile (#2018)

Signed-off-by: PeterWrighten <peterwrighten@gmail.com>
kubeflow · Mar 8, 2024 · 6133600 · 6133600
1 parent 57aa34d
commit 6133600
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/publish-example-images.yaml b/.github/workflows/publish-example-images.yaml
@@ -38,6 +38,8 @@ jobs:
             dockerfile: examples/pytorch/elastic/imagenet/Dockerfile
           - component-name: pytorch-elastic-example-echo
             dockerfile: examples/pytorch/elastic/echo/Dockerfile
+          - component-name: pytorch-cpu-demo-torchrun-cpu
+            dockerfile: examples/pytorch/cpu-demo/Dockerfile
           - component-name: xgboost-dist-iris
             dockerfile: examples/xgboost/xgboost-dist/Dockerfile
             context: examples/xgboost/xgboost-dist