publish torchrun example via Dockerfile

kubeflow · Mar 8, 2024 · 5e8f9e4 · 5e8f9e4
1 parent 57aa34d
commit 5e8f9e4
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