-
Notifications
You must be signed in to change notification settings - Fork 1
37 lines (31 loc) · 1.06 KB
/
build_wheel_cuda.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
name: Build python wheels - cuda
on:
workflow_dispatch
jobs:
build-wheels:
runs-on: ${{ matrix.os }}
strategy:
matrix:
os: [ubuntu-20.04]
steps:
- uses: actions/checkout@v2
# - uses: Jimver/[email protected]
# id: cuda-toolkit
# with:
# cuda: '12.5.0'
# log-file-suffix: '${{matrix.os}}.txt'
# linux-local-args: '["--toolkit"]'
- name: Build wheels
uses: pypa/[email protected]
with:
output-dir: dist
env:
CIBW_BUILD: cp38-manylinux_x86_64
CIBW_BUILD_VERBOSITY: 1
CIBW_BEFORE_ALL_LINUX: "wget https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-repo-rhel7-10.2.89-1.x86_64.rpm && rpm -i cuda-repo-*.rpm && yum install -y openblas-devel cuda"
CIBW_CONFIG_SETTINGS: "cmake.define.ENABLE_PYTHON_EMBED=OFF cmake.define.PYGENTEN_CUDA=ON"
- name: Upload distributions
uses: actions/upload-artifact@v4
with:
name: release-dists
path: dist/