Skip to content

Commit e3fbeaa

Browse files
committed
Updates Linux container and CUDA version
This commit updates Linux builds from the manylinux2014 container to the manylinux_2_28 container. The manylinux2014 container uses GCC 10.2, which produces code that runs substantially slower than more recent versions (~4 times in my tests). Code compiled on manylinux_2_28 should still run on most systems, so portability will not be greatly reduced. This commit also changes the version of the CUDA Toolkit used on the Linux builds from 11.6 to 12.4. This matches the Windows builds (where the more recent version was forced by the older version no longer being supported), and should result in the inclusion of code compiled for more recent GPU architectures.
1 parent 7c596d9 commit e3fbeaa

2 files changed

Lines changed: 7 additions & 7 deletions

File tree

.github/workflows/build.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,14 +25,14 @@ jobs:
2525
shell: bash
2626

2727
- name: Build wheels
28-
uses: pypa/cibuildwheel@v3.1.4
28+
uses: pypa/cibuildwheel@v3.3.0
2929
env:
30-
# For Linux, install CUDA in the container. Note that manylinux2014
30+
# For Linux, install CUDA in the container. Note that manylinux_2_28
3131
# is specified in pyproject.toml
3232
CIBW_BEFORE_ALL_LINUX: >
33-
yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo &&
34-
yum install -y cuda-nvcc-11-8 cuda-cudart-devel-11-8
35-
CIBW_ENVIRONMENT_LINUX: PATH="$PATH:/usr/local/cuda-11.8/bin" CUDA_PATH="/usr/local/cuda-11.8/bin" CUDA_HOME="/usr/local/cuda-11.8" CUDA_ROOT="/usr/local/cuda-11.8" LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64" CUDACXX="/usr/local/cuda-11.8/bin/nvcc"
33+
yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo &&
34+
yum install -y cuda-nvcc-12-4 cuda-cudart-devel-12-4
35+
CIBW_ENVIRONMENT_LINUX: PATH="$PATH:/usr/local/cuda-12.4/bin" CUDA_PATH="/usr/local/cuda-12.4/bin" CUDA_HOME="/usr/local/cuda-12.4" CUDA_ROOT="/usr/local/cuda-12.4" LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-12.4/lib64" CUDACXX="/usr/local/cuda-12.4/bin/nvcc"
3636
# For Windows, install CUDA
3737
CIBW_BEFORE_ALL_WINDOWS: >
3838
nuget install intelopenmp.devel.win -DirectDownload -NonInteractive &&

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@ build = "cp311-*"
3434
skip = ["*musllinux*"]
3535

3636
# Use an older manylinux image for better compatibility
37-
manylinux-x86_64-image = "manylinux2014"
38-
manylinux-aarch64-image = "manylinux2014"
37+
manylinux-x86_64-image = "manylinux_2_28"
38+
manylinux-aarch64-image = "manylinux_2_28"
3939

4040
build-verbosity = 1
4141

0 commit comments

Comments
 (0)