Skip to content

Support ARM64 platform in TensorFlow examples #2112

Closed
@tenzen-y

Description

@tenzen-y

Currently, we support only the AMD64 platform in all TensorFlow 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: https://github.com/kubeflow/training-operator/blob/e31d11faa9f6ce5111b60c01079d39295589e0ef/.github/workflows/publish-example-images.yaml#L24

  • examples/tensorflow/dist-mnist/Dockerfile
  • examples/tensorflow/distribution_strategy/estimator-API/Dockerfile
  • examples/tensorflow/distribution_strategy/keras-API/Dockerfile
  • examples/tensorflow/mnist_with_summaries/Dockerfile
  • examples/tensorflow/tf_sample/Dockerfile

/good-first-issue

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions