Skip to content

Commit bbe96e6

Browse files
committed
Add manylinux_2_28 cuda images
1 parent ead2e21 commit bbe96e6

File tree

2 files changed

+29
-0
lines changed

2 files changed

+29
-0
lines changed

.github/workflows/build-manywheel-images.yml

+21
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,27 @@ jobs:
6060
- name: Build Docker Image
6161
run: |
6262
manywheel/build_docker.sh
63+
build-docker-cuda-manylinux_2_28:
64+
runs-on: linux.12xlarge.ephemeral
65+
strategy:
66+
matrix:
67+
cuda_version: ["12.4", "12.1", "11.8"]
68+
env:
69+
GPU_ARCH_TYPE: cuda-manylinux_2_28
70+
GPU_ARCH_VERSION: ${{ matrix.cuda_version }}
71+
steps:
72+
- name: Purge tools folder (free space for build)
73+
run: rm -rf /opt/hostedtoolcache
74+
- name: Checkout PyTorch builder
75+
uses: actions/checkout@v3
76+
- name: Authenticate if WITH_PUSH
77+
run: |
78+
if [[ "${WITH_PUSH}" == true ]]; then
79+
echo "${DOCKER_TOKEN}" | docker login -u "${DOCKER_ID}" --password-stdin
80+
fi
81+
- name: Build Docker Image
82+
run: |
83+
manywheel/build_docker.sh
6384
build-docker-cuda-aarch64:
6485
runs-on: linux.arm64.2xlarge
6586
strategy:

manywheel/build_docker.sh

+8
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,14 @@ case ${GPU_ARCH_TYPE} in
6868
GPU_IMAGE=centos:7
6969
DOCKER_GPU_BUILD_ARG="--build-arg BASE_CUDA_VERSION=${GPU_ARCH_VERSION} --build-arg DEVTOOLSET_VERSION=9"
7070
;;
71+
cuda-manylinux_2_28)
72+
TARGET=cuda_final
73+
DOCKER_TAG=cuda${GPU_ARCH_VERSION}
74+
LEGACY_DOCKER_IMAGE=${DOCKER_REGISTRY}/pytorch/manylinux_2_28-cuda${GPU_ARCH_VERSION//./}
75+
GPU_IMAGE=amd64/almalinux:8
76+
DOCKER_GPU_BUILD_ARG="--build-arg BASE_CUDA_VERSION=${GPU_ARCH_VERSION} --build-arg DEVTOOLSET_VERSION=11"
77+
MANY_LINUX_VERSION="2_28"
78+
;;
7179
cuda-aarch64)
7280
TARGET=cuda_final
7381
DOCKER_TAG=cuda${GPU_ARCH_VERSION}

0 commit comments

Comments
 (0)