Skip to content

Commit fddb3e6

Browse files
authored
ci: Push manylinux docker image with cu132 (#610)
1 parent 98a6d5f commit fddb3e6

File tree

3 files changed

+9
-5
lines changed

3 files changed

+9
-5
lines changed
Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,14 @@
1-
FROM quay.io/pypa/manylinux_2_28_x86_64:2025.08.02-1 AS builder
1+
FROM quay.io/pypa/manylinux_2_28_x86_64:2026.03.20-1 AS builder
22
ARG CUDA_VERSION
33
COPY install_cuda.sh /tmp/
44
RUN dnf --setopt=install_weak_deps=False install -y aria2
55
RUN bash /tmp/install_cuda.sh ${CUDA_VERSION}
66

7-
FROM quay.io/pypa/manylinux_2_28_x86_64:2025.08.02-1
7+
FROM quay.io/pypa/manylinux_2_28_x86_64:2026.03.20-1
88
ARG CUDA_VERSION
99
COPY --from=builder /usr/local/cuda-${CUDA_VERSION} /usr/local/cuda-${CUDA_VERSION}
1010
# The base image comes with gcc 14. However, NVIDIA officially supports:
11+
# GCC<=15 for CUDA 13.2
1112
# GCC<=15 for CUDA 13.0
1213
# GCC<=14 for CUDA 12.9
1314
# GCC<=14 for CUDA 12.8
@@ -16,6 +17,6 @@ COPY --from=builder /usr/local/cuda-${CUDA_VERSION} /usr/local/cuda-${CUDA_VERSI
1617
# GCC<=12 for CUDA 12.1
1718
# GCC<=11 for CUDA 11.8
1819
# https://docs.nvidia.com/cuda/cuda-installation-guide-linux/#host-compiler-support-policy
19-
RUN dnf --setopt=install_weak_deps=False install -y gcc-toolset-11-toolchain
20+
RUN dnf --setopt=install_weak_deps=False install -y gcc-toolset-11-toolchain
2021
ENV PATH=/opt/rh/gcc-toolset-11/root/usr/bin:/usr/local/cuda-${CUDA_VERSION}/bin:${PATH}
2122
ENV LD_LIBRARY_PATH=/opt/rh/gcc-toolset-11/root/usr/lib64:/opt/rh/gcc-toolset-11/root/usr/lib:/usr/local/cuda-${CUDA_VERSION}/lib64:${LD_LIBRARY_PATH}

.github/workflows/docker/install_cuda.sh

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,10 @@ set -ex
44
CUDA_VERSION="${1:?Specify cuda version, e.g. 12.9}"
55

66
# See https://developer.nvidia.com/cuda-toolkit-archive
7-
if [ "$CUDA_VERSION" == "13.0" ]; then
7+
if [ "$CUDA_VERSION" == "13.2" ]; then
8+
URL=https://developer.download.nvidia.com/compute/cuda/13.2.0/local_installers/cuda-repo-rhel8-13-2-local-13.2.0_595.45.04-1.x86_64.rpm
9+
PACKAGE_NAME=cuda-toolkit-13-2
10+
elif [ "$CUDA_VERSION" == "13.0" ]; then
811
URL=https://developer.download.nvidia.com/compute/cuda/13.0.0/local_installers/cuda-repo-rhel8-13-0-local-13.0.0_580.65.06-1.x86_64.rpm
912
PACKAGE_NAME=cuda-toolkit-13-0
1013
elif [ "$CUDA_VERSION" == "12.9" ]; then
@@ -33,7 +36,6 @@ else
3336
fi
3437

3538
echo "Installing CUDA version: $CUDA_VERSION"
36-
# aria2c $URL -o /tmp/cuda.rpm
3739
aria2c -x 16 -s 16 -k 1M --file-allocation=none "$URL" -o /tmp/cuda.rpm
3840
rpm -i /tmp/cuda.rpm
3941
dnf clean all

CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
77

88
## [Unreleased]
99
### Added
10+
- Added CUDA 13.2 support to `manylinux` Dockerfile ([#614](https://github.com/pyg-team/pyg-lib/pull/614))
1011
### Changed
1112
### Deprecated
1213
### Removed

0 commit comments

Comments
 (0)