Update manylinux cuda image to 2_28 (#78)

* Update manylinux cuda image to 2_28

* Dont push on PR

.github/workflows/build-manylinux2014-cuda-aarch64.yml → .github/workflows/build-manylinux-cuda-aarch64.yml

@@ -1,16 +1,16 @@
-name: Build manylinux2014 aarch64 + cuda image
+name: Build manylinux aarch64 + cuda image
 
 on:
   push:
     branches:
       - main
     paths:
-      - 'manylinux2014_cuda_aarch64/*'
-      - '.github/workflows/build-manylinux2014-cuda-aarch64.yml'
+      - 'manylinux_cuda_aarch64/*'
+      - '.github/workflows/build-manylinux-cuda-aarch64.yml'
   pull_request:
     paths:
-      - 'manylinux2014_cuda_aarch64/*'
-      - '.github/workflows/build-manylinux2014-cuda-aarch64.yml'
+      - 'manylinux_cuda_aarch64/*'
+      - '.github/workflows/build-manylinux-cuda-aarch64.yml'
   workflow_dispatch:
   schedule:
     - cron: '30 12 * * 5'
@@ -25,7 +25,8 @@ jobs:
     uses: Chia-Network/actions/.github/workflows/docker-build.yaml@main
     with:
       docker-platforms: linux/arm64
-      docker-context: ./manylinux2014_cuda_aarch64
-      dockerfile: ./manylinux2014_cuda_aarch64/Dockerfile
-      image_subpath: manylinux2014_cuda_aarch64
+      docker-context: ./manylinux_cuda_aarch64
+      dockerfile: ./manylinux_cuda_aarch64/Dockerfile
+      image_subpath: manylinux_cuda_aarch64
       alternate-latest-mode: true
+      push: ${{ github.event_name != 'pull_request' }}

.github/workflows/build-manylinux2014-cuda-x86_64.yml → .github/workflows/build-manylinux-cuda-x86_64.yml

@@ -1,16 +1,16 @@
-name: Build manylinux2014 x86_64 + cuda image
+name: Build manylinux x86_64 + cuda image
 
 on:
   push:
     branches:
       - main
     paths:
-      - 'manylinux2014_cuda_x86_64/*'
-      - '.github/workflows/build-manylinux2014-cuda-x86_64.yml'
+      - 'manylinux_cuda_x86_64/*'
+      - '.github/workflows/build-manylinux-cuda-x86_64.yml'
   pull_request:
     paths:
-      - 'manylinux2014_cuda_x86_64/*'
-      - '.github/workflows/build-manylinux2014-cuda-x86_64.yml'
+      - 'manylinux_cuda_x86_64/*'
+      - '.github/workflows/build-manylinux-cuda-x86_64.yml'
   workflow_dispatch:
   schedule:
     - cron: '30 12 * * 5'
@@ -25,7 +25,8 @@ jobs:
     uses: Chia-Network/actions/.github/workflows/docker-build.yaml@main
     with:
       docker-platforms: linux/amd64
-      docker-context:  ./manylinux2014_cuda_x86_64
-      dockerfile: ./manylinux2014_cuda_x86_64/Dockerfile
-      image_subpath: manylinux2014_cuda_x86_64
+      docker-context:  ./manylinux_cuda_x86_64
+      dockerfile: ./manylinux_cuda_x86_64/Dockerfile
+      image_subpath: manylinux_cuda_x86_64
       alternate-latest-mode: true
+      push: ${{ github.event_name != 'pull_request' }}

manylinux_cuda_aarch64/Dockerfile

@@ -0,0 +1,10 @@
+FROM quay.io/pypa/manylinux_2_28_aarch64
+
+ENV PATH=/usr/local/cuda-12.5/bin${PATH:+:${PATH}}
+
+RUN dnf -y install https://dl.fedoraproject.org/pub/epel/epel-release-latest-8.noarch.rpm && \
+    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 && \
+    dnf clean all && \
+    dnf install -y cuda-toolkit

manylinux_cuda_x86_64/Dockerfile

@@ -0,0 +1,9 @@
+FROM quay.io/pypa/manylinux_2_28_x86_64
+
+ENV PATH=/usr/local/cuda-12.5/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