[ci][py11/2] support docker build image for arm64 (#36586)

ray-project · Jun 26, 2023 · 2f84fa3 · 2f84fa3
1 parent 82f96d6
commit 2f84fa3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.buildkite/pipeline.arm64.yml b/.buildkite/pipeline.arm64.yml
@@ -48,6 +48,7 @@
     - py38
     - py39
     - py310
+    - py311
 
 - label: ":mechanical_arm: :docker: Build Images: {{matrix}} [aarch64] cu117/cu118"
   conditions: ["RAY_CI_PYTHON_DEPENDENCIES_AFFECTED", "RAY_CI_DOCKER_AFFECTED", "RAY_CI_CORE_CPP_AFFECTED"]
@@ -63,3 +64,4 @@
     - py38
     - py39
     - py310
+    - py311
diff --git a/docker/base-deps/Dockerfile b/docker/base-deps/Dockerfile
@@ -76,7 +76,7 @@ RUN sudo apt-get update -y && sudo apt-get upgrade -y \
     # We install cmake temporarily to get psutil
     && sudo apt-get autoremove -y cmake zlib1g-dev \
         # We keep g++ on GPU images, because uninstalling removes CUDA Devel tooling
-        $(if [ "$BASE_IMAGE" = "ubuntu:focal" ]; then echo \
+        $(if [[ "$BASE_IMAGE" == "ubuntu:focal" && "$HOSTTYPE" == "x86_64" ]]; then echo \
         g++; fi) \
     # Either install kubectl or remove wget 
     && (if [ "$AUTOSCALER" = "autoscaler" ]; \
-Original file line number
+Diff line change
@@ Expand Up / @@ -48,6 +48,7 @@ @@
         - py38
         - py39
         - py310
+        - py311
     - label: ":mechanical_arm: :docker: Build Images: {{matrix}} [aarch64] cu117/cu118"
       conditions: ["RAY_CI_PYTHON_DEPENDENCIES_AFFECTED", "RAY_CI_DOCKER_AFFECTED", "RAY_CI_CORE_CPP_AFFECTED"]
@@ Expand All / @@ -63,3 +64,4 @@ @@
         - py38
         - py39
         - py310
+        - py311