diff --git a/.github/workflows/build-manywheel-images.yml b/.github/workflows/build-manywheel-images.yml index d88fcbedd..333f3de5d 100644 --- a/.github/workflows/build-manywheel-images.yml +++ b/.github/workflows/build-manywheel-images.yml @@ -60,6 +60,27 @@ jobs: - name: Build Docker Image run: | manywheel/build_docker.sh + build-docker-cuda-manylinux_2_28: + runs-on: linux.12xlarge.ephemeral + strategy: + matrix: + cuda_version: ["12.4", "12.1", "11.8"] + env: + GPU_ARCH_TYPE: cuda-manylinux_2_28 + GPU_ARCH_VERSION: ${{ matrix.cuda_version }} + steps: + - name: Purge tools folder (free space for build) + run: rm -rf /opt/hostedtoolcache + - name: Checkout PyTorch builder + uses: actions/checkout@v3 + - name: Authenticate if WITH_PUSH + run: | + if [[ "${WITH_PUSH}" == true ]]; then + echo "${DOCKER_TOKEN}" | docker login -u "${DOCKER_ID}" --password-stdin + fi + - name: Build Docker Image + run: | + manywheel/build_docker.sh build-docker-cuda-aarch64: runs-on: linux.arm64.2xlarge strategy: diff --git a/manywheel/build_docker.sh b/manywheel/build_docker.sh index 5c35d32ae..819a4a003 100755 --- a/manywheel/build_docker.sh +++ b/manywheel/build_docker.sh @@ -68,6 +68,14 @@ case ${GPU_ARCH_TYPE} in GPU_IMAGE=centos:7 DOCKER_GPU_BUILD_ARG="--build-arg BASE_CUDA_VERSION=${GPU_ARCH_VERSION} --build-arg DEVTOOLSET_VERSION=9" ;; + cuda-manylinux_2_28) + TARGET=cuda_final + DOCKER_TAG=cuda${GPU_ARCH_VERSION} + LEGACY_DOCKER_IMAGE=${DOCKER_REGISTRY}/pytorch/manylinux_2_28-cuda${GPU_ARCH_VERSION//./} + GPU_IMAGE=amd64/almalinux:8 + DOCKER_GPU_BUILD_ARG="--build-arg BASE_CUDA_VERSION=${GPU_ARCH_VERSION} --build-arg DEVTOOLSET_VERSION=11" + MANY_LINUX_VERSION="2_28" + ;; cuda-aarch64) TARGET=cuda_final DOCKER_TAG=cuda${GPU_ARCH_VERSION}