mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-15 04:52:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix building cuda wheels

Browse Source
This commit is contained in:
Fangjun Kuang 2024-02-10 14:05:17 +08:00
parent b75abef4c0
commit 27c184e89a
2 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
.github/workflows/ubuntu-cuda-wheels.yml vendored
View File

@ -44,6 +44,15 @@ jobs:
with: with:
fetch-depth: 0 fetch-depth: 0
- name: Free space
shell: bash
run: |
df -h
rm -rf /opt/hostedtoolcache
df -h
echo "pwd: $PWD"
echo "github.workspace ${{ github.workspace }}"
# see https://github.com/pytorch/test-infra/blob/9e3d392690719fac85bad0c9b67f530e48375ca1/tools/scripts/generate_binary_build_matrix.py # see https://github.com/pytorch/test-infra/blob/9e3d392690719fac85bad0c9b67f530e48375ca1/tools/scripts/generate_binary_build_matrix.py
# https://github.com/pytorch/builder/tree/main/manywheel # https://github.com/pytorch/builder/tree/main/manywheel
# https://github.com/pytorch/builder/pull/476 # https://github.com/pytorch/builder/pull/476

8
scripts/github_actions/generate_build_matrix.py
View File

@ -53,10 +53,10 @@ def generate_build_matrix(enable_cuda, for_windows, for_macos, test_only_latest_
# "python-version": ["3.6", "3.7", "3.8"], # "python-version": ["3.6", "3.7", "3.8"],
# "cuda": ["10.1", "10.2"], # "cuda": ["10.1", "10.2"],
# }, # },
"1.6.0": { # "1.6.0": {
"python-version": ["3.6", "3.7", "3.8"], # "python-version": ["3.6", "3.7", "3.8"],
"cuda": ["10.1", "10.2"] if not for_windows else ["10.1.243", "10.2.89"], # "cuda": ["10.1", "10.2"] if not for_windows else ["10.1.243", "10.2.89"],
}, # },
"1.7.0": { "1.7.0": {
"python-version": ["3.6", "3.7", "3.8"], "python-version": ["3.6", "3.7", "3.8"],
"cuda": ["10.1", "10.2", "11.0"] "cuda": ["10.1", "10.2", "11.0"]