Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[docker/arm64] Build docker images (with suffix) for arm64 #31861

Merged
merged 17 commits into from
Jan 26, 2023

Conversation

krfricke
Copy link
Contributor

@krfricke krfricke commented Jan 23, 2023

Why are these changes needed?

This PR adds Docker image builds for arm64 (aarch64). The images will be tagged with a suffix -aarch64. In a future retagging step, we can combine the existing x86_64 images with the aarch64 images for true multiplatform images.

Changes needed to make this work:

  • Move libgcc to libgcc-ng because the former does not exist on conda for aarch64
  • Upgrade Python 3.7.7 to 3.7.16 because the former does not exist on conda for aarch74

Please also note that aarch64 base images only exist for CUDA >= 11.2.

Lastly, we are not building ray-ml docker images. This is because not all current dependencies are compatible with aarch64 - specifically, tensorflow requires at least 1.10.1 and torch 1.13.0 does not seem to be released for linux aarch64.

Related issue number

Closes #31537

Checks

  • I've signed off every commit(by using the -s flag, i.e., git commit -s) in this PR.
  • I've run scripts/format.sh to lint the changes in this PR.
  • I've included any doc changes needed for https://docs.ray.io/en/master/.
  • I've made sure the tests are passing. Note that there might be a few flaky tests, see the recent failures at https://flakey-tests.ray.io/
  • Testing Strategy
    • Unit tests
    • Release tests
    • This PR is not tested :(

Kai Fricke added 11 commits January 23, 2023 10:25
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
@krfricke krfricke marked this pull request as ready for review January 25, 2023 00:36
@krfricke krfricke requested a review from amogkam as a code owner January 25, 2023 00:36
@krfricke krfricke requested a review from richardliaw January 25, 2023 00:36
Kai Fricke added 2 commits January 25, 2023 15:43
Copy link
Contributor

@amogkam amogkam left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm! just one minor comment

@@ -636,6 +706,11 @@ def push_readmes(merge_build: bool):
required=True,
help="Whether to bypass checking if docker is affected",
)
parser.add_argument(
"--suffix",
required=False,
Copy link
Contributor

@amogkam amogkam Jan 26, 2023

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we can specify choices here to avoid any mistakes when specifying the command in the buildkite pipeline

Suggested change
required=False,
required=False,
choices=["aarm64"]

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Good point, I'll add this with a constant. Thanks!

Kai Fricke added 2 commits January 26, 2023 09:36
Kai Fricke added 2 commits January 26, 2023 10:07
Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Kai Fricke <kai@anyscale.com>
@krfricke krfricke merged commit 994c3d6 into ray-project:master Jan 26, 2023
@krfricke krfricke deleted the docker/aarch64 branch January 26, 2023 19:19
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[docker] Add docker images for linux/aarch64 platform
3 participants