From 770e562a37d4af82fe42224cca48b0b983965e7d Mon Sep 17 00:00:00 2001 From: William Allen Date: Tue, 4 Feb 2025 17:33:34 -0600 Subject: [PATCH] Update build-images to fix bladebit CI (#94) * fix old glibc * more accurate commands * correct some things * Update Dockerfile --- manylinux_cuda_aarch64/Dockerfile | 7 ++++--- manylinux_cuda_x86_64/Dockerfile | 14 +++++++------- 2 files changed, 11 insertions(+), 10 deletions(-) diff --git a/manylinux_cuda_aarch64/Dockerfile b/manylinux_cuda_aarch64/Dockerfile index 3cdb25b..b799477 100644 --- a/manylinux_cuda_aarch64/Dockerfile +++ b/manylinux_cuda_aarch64/Dockerfile @@ -1,8 +1,9 @@ -FROM quay.io/pypa/manylinux_2_28_aarch64 +FROM quay.io/pypa/manylinux_2_34_aarch64 -ENV PATH=/usr/local/cuda-12.5/bin${PATH:+:${PATH}} +ENV PATH=/usr/local/cuda/bin${PATH:+:${PATH}} -RUN dnf -y install https://dl.fedoraproject.org/pub/epel/epel-release-latest-8.noarch.rpm && \ +RUN dnf update -y && \ + dnf install -y epel-release && \ dnf install -y kernel-headers.aarch64 kernel-devel.aarch64 && \ dnf config-manager --add-repo http://developer.download.nvidia.com/compute/cuda/repos/rhel9/aarch64/cuda-rhel9.repo && \ dnf config-manager --add-repo http://developer.download.nvidia.com/compute/cuda/repos/rhel9/sbsa/cuda-rhel9.repo && \ diff --git a/manylinux_cuda_x86_64/Dockerfile b/manylinux_cuda_x86_64/Dockerfile index 6946ef9..5768201 100644 --- a/manylinux_cuda_x86_64/Dockerfile +++ b/manylinux_cuda_x86_64/Dockerfile @@ -1,9 +1,9 @@ -FROM quay.io/pypa/manylinux_2_28_x86_64 +FROM quay.io/pypa/manylinux_2_34_x86_64 -ENV PATH=/usr/local/cuda-12.5/bin${PATH:+:${PATH}} +ENV PATH=/usr/local/cuda/bin${PATH:+:${PATH}} -RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo && \ - dnf -y install https://dl.fedoraproject.org/pub/epel/epel-release-latest-8.noarch.rpm && \ - dnf clean all && \ - dnf -y module install nvidia-driver:latest-dkms && \ - dnf -y install cuda-toolkit +RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel9/x86_64/cuda-rhel9.repo && \ + dnf install -y epel-release && \ + dnf clean expire-cache && \ + dnf module enable -y nvidia-driver:560-open && \ + dnf install -y cuda