Update CIBW.

.gitlab-ci.yml

@@ -12,23 +12,23 @@ stages:
   - deploy
 
 include:
-  - project: 'eclipse/aidge/gitlab_shared_files' 
+  - project: 'eclipse/aidge/gitlab_shared_files'
     ref: 'main'
-    file: 
+    file:
       # choose which jobs to run by including the corresponding files.
       - '.gitlab/ci/ubuntu_cpp.gitlab-ci.yml'
 
       - '.gitlab/ci/ubuntu_python.gitlab-ci.yml'
-      - '.gitlab/ci/release/cibuildwheel_ubuntu.gitlab-ci.yml'   
+      - '.gitlab/ci/release/cibuildwheel_ubuntu.gitlab-ci.yml'
 
       # - '.gitlab/ci/windows_cpp.gitlab-ci.yml'
 
-      # - '.gitlab/ci/windows_python.gitlab-ci.yml'   
-      # - '.gitlab/ci/release/cibuildwheel_windows.gitlab-ci.yml'   
+      # - '.gitlab/ci/windows_python.gitlab-ci.yml'
+      # - '.gitlab/ci/release/cibuildwheel_windows.gitlab-ci.yml'
 
 
 release:pip:ubuntu:
-  tags: 
+  tags:
     - release:cuda
   variables:
     DOCKER_HOST: unix:///var/run/docker.sock
@@ -47,8 +47,8 @@ release:pip:ubuntu:
 
   parallel:
     matrix:
-      - CIBW_BUILD: "cp38-manylinux_x86_64"
-      - CIBW_BUILD: "cp39-manylinux_x86_64"
+      # - CIBW_BUILD: "cp38-manylinux_x86_64"
+      # - CIBW_BUILD: "cp39-manylinux_x86_64"
       - CIBW_BUILD: "cp310-manylinux_x86_64"
       - CIBW_BUILD: "cp311-manylinux_x86_64"
       - CIBW_BUILD: "cp312-manylinux_x86_64"
@@ -60,7 +60,8 @@ release:pip:ubuntu:
     - !reference [.ubuntu:download:repositories, before_script] # located in common.gitlab-ci.yml
 
   script:
-    - /home/ubuntu/.local/bin/cibuildwheel --output-dir wheelhouse
+    # - /home/ubuntu/.local/bin/cibuildwheel --output-dir wheelhouse
+    - cibuildwheel --output-dir wheelhouse
 
   after_script:
     # Ensure all files are owned by the correct user at the end of the job

pyproject.toml

@@ -45,12 +45,13 @@ build-backend = "setuptools.build_meta"
 build-frontend = "build"
 test-requires = "pytest"
 test-command = "pytest {project}/aidge_backend_cuda/unit_tests"
+manylinux-x86_64-image = "manylinux_2_28"
 # uncomment to run cibuildwheel locally on selected distros
-# build=[
+build=[
 #     "cp38-manylinux_x86_64",
 #     "cp39-manylinux_x86_64",
-#     "cp310-manylinux_x86_64"
-# ]
+    "cp310-manylinux_x86_64"
+]
 
 repair-wheel-command = "auditwheel --verbose repair {wheel} -w {dest_dir} --exclude libcudart.so.12 --exclude libcudnn.so.9 --exclude libcublas.so.12 --exclude libcublasLt.so.12"
 
@@ -58,7 +59,7 @@ repair-wheel-command = "auditwheel --verbose repair {wheel} -w {dest_dir} --excl
 # pass command line options to 'docker run'
 name = "docker"
 create-args = [
-    "--runtime=nvidia",
+    # "--runtime=nvidia",
     "--gpus", "all",
     "--privileged",
     "-v","/cache",
@@ -82,19 +83,24 @@ LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
 PATH="/usr/local/cuda/bin:$PATH"
 [tool.cibuildwheel.linux]
 before-build = [
+    "export rhel_version=$(grep -oP '(?<=^VERSION_ID=\")[0-9]+' /etc/os-release)",
     "export CUDA_TOOLKIT_VERSION=$CUDA_MAJOR_VERSION-$CUDA_MINOR_VERSION",
-    "echo '\n\n\n\n yum -y install cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH'",
-    "yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/$ARCH/cuda-rhel7.repo",
-    "yum clean all",
-    "yum install -y --nogpgcheck cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH",
-    "yum list available | grep cudnn",
-    "yum install -y --nogpgcheck libcudnn$CUDNN_VERSION-cuda-$CUDA_MAJOR_VERSION.$ARCH",
-    "yum install -y --nogpgcheck libcudnn$CUDNN_VERSION-devel-cuda-$CUDA_MAJOR_VERSION.$ARCH",
+
+    "dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel$rhel_version/$ARCH/cuda-rhel$rhel_version.repo",
+    "dnf clean all",
+
+    "dnf install -y --nogpgcheck cuda-toolkit-$CUDA_TOOLKIT_VERSION.$ARCH",
+    "dnf list available | grep cudnn",
+
+    "dnf install -y --nogpgcheck libcudnn$CUDNN_VERSION-cuda-$CUDA_MAJOR_VERSION.$ARCH",
+    "dnf install -y --nogpgcheck libcudnn$CUDNN_VERSION-devel-cuda-$CUDA_MAJOR_VERSION.$ARCH",
+
     "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH",
     "export PATH=/usr/local/cuda/bin:$PATH",
+
     "which nvcc",
     "nvcc --version",
-    "echo '\n\n\n\nInstalling required dependencies for aidge_backend_cuda.\n\n'",
+
     "echo '\n\n\n\nInstalling required dependencies for aidge_backend_cuda.\n\n'",
     "bash .gitlab/ci/cibuildwheel_build_deps_before_build_wheel.sh /host/$SEARCH_PATH"
 ]