File tree 2 files changed +29
-0
lines changed
2 files changed +29
-0
lines changed Original file line number Diff line number Diff line change 60
60
- name : Build Docker Image
61
61
run : |
62
62
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
63
84
build-docker-cuda-aarch64 :
64
85
runs-on : linux.arm64.2xlarge
65
86
strategy :
Original file line number Diff line number Diff line change @@ -68,6 +68,14 @@ case ${GPU_ARCH_TYPE} in
68
68
GPU_IMAGE=centos:7
69
69
DOCKER_GPU_BUILD_ARG=" --build-arg BASE_CUDA_VERSION=${GPU_ARCH_VERSION} --build-arg DEVTOOLSET_VERSION=9"
70
70
;;
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
+ ;;
71
79
cuda-aarch64)
72
80
TARGET=cuda_final
73
81
DOCKER_TAG=cuda${GPU_ARCH_VERSION}
You can’t perform that action at this time.
0 commit comments