Support ARM64 platform in PyTorch examples about training-operator

[tenzen-y]

Hey! Interested in this project, could I be assigned to this issue?

Sure, feel free to assign yourself with /assign comment.

[google-oss-prow]

@tenzen-y:
This request has been marked as suitable for new contributors.

Please ensure the request meets the requirements listed here.

If this request no longer meets these requirements, the label can be removed
by commenting with the /remove-good-first-issue command.

In response to this:

Currently, we support only the AMD64 platform in all PyTorch examples.
However, some users use ARM64 machines such as Macbooks. So, supporting the ARM64 platform allows users to try the training-operator examples on their machines easily.

Contributors should support the following examples by adding the platform parameters here:

training-operator/.github/workflows/publish-example-images.yaml

Line 24 in e31d11f

include:

• examples/pytorch/cpu-demo/Dockerfile
• examples/pytorch/elastic/imagenet/Dockerfile
• examples/pytorch/elastic/echo/Dockerfile
• examples/pytorch/mnist/Dockerfile
• examples/pytorch/mnist/Dockerfile-mpi
• examples/pytorch/smoke-dist/Dockerfile

/good-first-issue

Instructions for interacting with me using PR comments are available here. If you have questions or suggestions related to my behavior, please file an issue against the kubernetes-sigs/prow repository.

[danielsuh05]

Hey! Interested in this project, could I be assigned to this issue?

[danielsuh05]

/assign