-
Notifications
You must be signed in to change notification settings - Fork 5.9k
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
Conversation
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>
There was a problem hiding this 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, |
There was a problem hiding this comment.
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
required=False, | |
required=False, | |
choices=["aarm64"] |
There was a problem hiding this comment.
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!
Signed-off-by: Kai Fricke <kai@anyscale.com>
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:
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
git commit -s
) in this PR.scripts/format.sh
to lint the changes in this PR.