File tree Expand file tree Collapse file tree 3 files changed +9
-5
lines changed
Expand file tree Collapse file tree 3 files changed +9
-5
lines changed Original file line number Diff line number Diff line change 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
22ARG CUDA_VERSION
33COPY install_cuda.sh /tmp/
44RUN dnf --setopt=install_weak_deps=False install -y aria2
55RUN 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
88ARG CUDA_VERSION
99COPY --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
2021ENV PATH=/opt/rh/gcc-toolset-11/root/usr/bin:/usr/local/cuda-${CUDA_VERSION}/bin:${PATH}
2122ENV 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}
Original file line number Diff line number Diff line change 44CUDA_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
1013elif [ " $CUDA_VERSION " == " 12.9" ]; then
3336fi
3437
3538echo " Installing CUDA version: $CUDA_VERSION "
36- # aria2c $URL -o /tmp/cuda.rpm
3739aria2c -x 16 -s 16 -k 1M --file-allocation=none " $URL " -o /tmp/cuda.rpm
3840rpm -i /tmp/cuda.rpm
3941dnf clean all
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments