This repository was archived by the owner on Aug 15, 2025. It is now read-only.
File tree Expand file tree Collapse file tree 2 files changed +29
-0
lines changed Expand file tree Collapse file tree 2 files changed +29
-0
lines changed Original file line number Diff line number Diff line change 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 :
Original file line number Diff line number Diff 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}
You can’t perform that action at this time.
0 commit comments