Support arm64 in darts-cnn-cifar10 image

[tenzen-y]

Signed-off-by: tenzen-y yuki.iwai.tz@gmail.com

What this PR does / why we need it:
I modified Dockerfile for darts-cnn-cifar10 to support arm64 and I have tested in the below commands:

• linux/arm64 with GPU support

docker run -it --platform linux/arm64 ytenzen/multiplatform-darts-cnn-cifar10-gpu:latest --algorithm-settings="{'num_epochs': '1', 'w_lr': 0.025, 'w_lr_min': 0.001, 'w_momentum': 0.9, 'w_weight_decay': 0.0003, 'w_grad_clip': 5.0, 'alpha_lr': 0.0003, 'alpha_weight_decay': 0.001, 'batch_size': 128, 'num_workers': 0, 'init_channels': '1', 'print_step': 50, 'num_nodes': '1', 'stem_multiplier': '1'}" --search-space="['max_pooling_3x3']" --num-layers="1"

• linux/arm64 with CPU support

docker run -it --platform linux/arm64 ytenzen/multiplatform-darts-cnn-cifar10-cpu:latest --algorithm-settings="{'num_epochs': '1', 'w_lr': 0.025, 'w_lr_min': 0.001, 'w_momentum': 0.9, 'w_weight_decay': 0.0003, 'w_grad_clip': 5.0, 'alpha_lr': 0.0003, 'alpha_weight_decay': 0.001, 'batch_size': 128, 'num_workers': 0, 'init_channels': '1', 'print_step': 50, 'num_nodes': '1', 'stem_multiplier': '1'}" --search-space="['max_pooling_3x3']" --num-layers="1"

• linux/amd64 with GPU support

docker run -it --platform linux/amd64 ytenzen/multiplatform-darts-cnn-cifar10-gpu:latest --algorithm-settings="{'num_epochs': '1', 'w_lr': 0.025, 'w_lr_min': 0.001, 'w_momentum': 0.9, 'w_weight_decay': 0.0003, 'w_grad_clip': 5.0, 'alpha_lr': 0.0003, 'alpha_weight_decay': 0.001, 'batch_size': 128, 'num_workers': 0, 'init_channels': '1', 'print_step': 50, 'num_nodes': '1', 'stem_multiplier': '1'}" --search-space="['max_pooling_3x3']" --num-layers="1"

• linux/amd64 with CPU support

docker run -it --platform linux/amd64 ytenzen/multiplatform-darts-cnn-cifar10-cpu:latest --algorithm-settings="{'num_epochs': '1', 'w_lr': 0.025, 'w_lr_min': 0.001, 'w_momentum': 0.9, 'w_weight_decay': 0.0003, 'w_grad_clip': 5.0, 'alpha_lr': 0.0003, 'alpha_weight_decay': 0.001, 'batch_size': 128, 'num_workers': 0, 'init_channels': '1', 'print_step': 50, 'num_nodes': '1', 'stem_multiplier': '1'}" --search-space="['max_pooling_3x3']" --num-layers="1"

Which issue(s) this PR fixes (optional, in fixes #<issue number>(, fixes #<issue_number>, ...) format, will close the issue(s) when PR gets merged):
Part-of #1900

Checklist:

• Docs included if any changes are user facing

[coveralls]

Coverage increased (+0.1%) to 73.545% when pulling 12a9770 on tenzen-y:support-arm64-in-darts-cnn-cifar10 into daf5b9b on kubeflow:master.

[anencore94]

/lgtm
I see there is only one image left now. Thanks for your efforts

[johnugeorge]

/approve

[google-oss-prow]

[APPROVALNOTIFIER] This PR is APPROVED

This pull-request has been approved by: johnugeorge, tenzen-y

The full list of commands accepted by this bot can be found here.

The pull request process is described here

Needs approval from an approver in each of these files:

• OWNERS [johnugeorge]

Approvers can indicate their approval by writing /approve in a comment
Approvers can cancel approval by writing /approve cancel in a comment