Skip to content

Commit

Permalink
Update wheels.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
faberno authored Oct 4, 2024
1 parent 8f5c92e commit a3647cb
Showing 1 changed file with 11 additions and 10 deletions.
21 changes: 11 additions & 10 deletions .github/workflows/wheels.yml
Original file line number Diff line number Diff line change
Expand Up @@ -48,21 +48,22 @@ jobs:
- name: Install cibuildwheel
run: python -m pip install cibuildwheel

- name: Build CUDA Linux Wheels
run: python -m cibuildwheel --output-dir wheelhouse/cuda
if: runner.os == 'Linux'
- name: Build cuda wheels
env:
CIBW_BUILD_VERBOSITY: 1
CIBW_ARCHS_LINUX: "x86_64"
CIBW_SKIP: "*musl*"
CIBW_SKIP: "pp* *-manylinux_i686 *-musllinux_*"
CIBW_BEFORE_ALL: >
yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo &&
yum clean all &&
yum -y install cuda-toolkit &&
git submodule update --init --recursive &&
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin &&
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 &&
wget https://developer.download.nvidia.com/compute/cuda/12.5.0/local_installers/cuda-repo-ubuntu2004-12-5-local_12.5.0-555.42.02-1_amd64.deb &&
sudo dpkg -i cuda-repo-ubuntu2004-12-5-local_12.5.0-555.42.02-1_amd64.deb &&
sudo cp /var/cuda-repo-ubuntu2004-12-5-local/cuda-*-keyring.gpg /usr/share/keyrings/ &&
sudo apt-get update &&
sudo apt-get -y install cuda-toolkit-12-5 &&
ls -al /usr/local &&
export PATH=$PATH:/usr/local/cuda/bin &&
nvcc --version &&
export BUILD_WITH_CUDA=1
nvcc --version
# - uses: actions/setup-python@v5
# with:
Expand Down

0 comments on commit a3647cb

Please sign in to comment.