Skip to content

Commit 4bf007a

Browse files
GPU in wheels, attempt 1
1 parent 9991bc5 commit 4bf007a

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

.github/workflows/testing_wheels.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,11 @@ jobs:
5252
# Used to host cibuildwheel
5353
- uses: actions/setup-python@v2
5454

55+
# Install NVCC on Ubuntu to include GPU in the wheel.
56+
- name: cuda-toolkit
57+
if: ${{ matrix.os == 'ubuntu-20.04' }}
58+
uses: Jimver/[email protected]
59+
5560
- name: Install cibuildwheel and twine
5661
run: python -m pip install cibuildwheel==1.11.0
5762

0 commit comments

Comments
 (0)