[ci] use LightGBM CI image for building aarch64 wheels (fixes #5595) (#5622)

.ci/setup.sh

@@ -98,34 +98,6 @@ else  # Linux
             sudo apt-get update
             sudo apt-get install --no-install-recommends -y \
                 pocl-opencl-icd
-        elif [[ $(uname -m) == "aarch64" ]]; then
-            yum install -y \
-                epel-release \
-                gcc-c++ \
-                hwloc-devel \
-                sudo
-            yum install -y \
-                llvm-toolset-7.0-clang-devel \
-                llvm-toolset-7.0-llvm-devel \
-                ocl-icd-devel
-            git clone --depth 1 --branch v1.8 https://github.com/pocl/pocl.git
-            cmake \
-              -B pocl/build \
-              -S pocl \
-              -DCMAKE_BUILD_TYPE=release \
-              -DCMAKE_C_COMPILER=/usr/bin/gcc \
-              -DCMAKE_CXX_COMPILER=/usr/bin/g++ \
-              -DCMAKE_C_FLAGS=-std=gnu99 \
-              -DPOCL_INSTALL_ICD_VENDORDIR=/etc/OpenCL/vendors \
-              -DPOCL_DEBUG_MESSAGES=OFF \
-              -DINSTALL_OPENCL_HEADERS=OFF \
-              -DENABLE_SPIR=OFF \
-              -DENABLE_POCLCC=OFF \
-              -DENABLE_TESTS=OFF \
-              -DENABLE_EXAMPLES=OFF \
-              -DLLC_HOST_CPU=generic
-            cmake --build pocl/build -j4
-            sudo cmake --install pocl/build
         elif [[ $(uname -m) == "x86_64" ]]; then
             sudo yum update -y
             sudo yum install -y \

.vsts-ci.yml

@@ -192,7 +192,7 @@ jobs:
       $ROOT_DOCKER_FOLDER/.ci/setup.sh || exit -1
       $ROOT_DOCKER_FOLDER/.ci/test.sh || exit -1
       EOF
-      IMAGE_URI="quay.io/pypa/manylinux2014_${ARCH}"
+      IMAGE_URI="lightgbm/vsts-agent:manylinux2014_aarch64"
       docker pull "${IMAGE_URI}" || exit -1
       PLATFORM=$(docker inspect --format='{{.Os}}/{{.Architecture}}' "${IMAGE_URI}") || exit -1
       echo "detected image platform: ${PLATFORM}"