mirrors/kaldifeat
1
0
Fork 0
mirror of https://github.com/csukuangfj/kaldifeat.git synced 2025-08-19 14:56:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2022-06-03 17:40:35 +08:00
parent 1f4d4c7e43
commit 1b9c2a21e1
2 changed files with 24 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
.github/workflows/nightly-tests-windows-pip-cuda.yml vendored
View File

@ -19,7 +19,7 @@ name: Nightly tests windows pip cuda
on: on:
push: push:
branches: branches:
- nightly-test-2 - nightly-test
schedule: schedule:
# minute (0-59) # minute (0-59)
@ -44,8 +44,8 @@ jobs:
id: set-matrix id: set-matrix
run: | run: |
# outputting for debugging purposes # outputting for debugging purposes
python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows --test-only-latest-torch
MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows) MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows --test-only-latest-torch)
echo "::set-output name=matrix::${MATRIX}" echo "::set-output name=matrix::${MATRIX}"
run_nightly_tests_windows_pip_cuda: run_nightly_tests_windows_pip_cuda:
@ -75,7 +75,7 @@ jobs:
# See https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts # See https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
# for available CUDA versions # for available CUDA versions
- uses: Jimver/cuda-toolkit@v0.2.5 - uses: Jimver/cuda-toolkit@v0.2.7
id: cuda-toolkit id: cuda-toolkit
with: with:
cuda: ${{ matrix.cuda }} cuda: ${{ matrix.cuda }}
@ -86,6 +86,21 @@ jobs:
echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}" echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
nvcc --version nvcc --version
- name: Download cuDNN
shell: bash
run: |
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/csukuangfj/cudnn-for-windows
cd cudnn-for-windows
git lfs pull --include="cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive.zip"
unzip cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive.zip
ls -lh *
ls -lh */*
cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive
sed -i '59i#define CUDNN_MAJOR 8' cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive/include/cudnn.h
echo "PWD: $PWD"
- name: Install PyTorch ${{ matrix.torch }} - name: Install PyTorch ${{ matrix.torch }}
shell: bash shell: bash
run: | run: |
@ -105,6 +120,7 @@ jobs:
- name: Install from PyPI - name: Install from PyPI
shell: bash shell: bash
run: | run: |
export KALDIFEAT_CMAKE_ARGS="-DCMAKE_BUILD_TYPE=$BUILD_TYPE -DCUDNN_INCLUDE_PATH=/d/a/kaldifeat/kaldifeat/cudnn-for-windows/cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive/include -DCUDNN_LIBRARY_PATH=/d/a/kaldifeat/kaldifeat/cudnn-for-windows/cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive/lib/cudnn.lib"
export KALDIFEAT_MAKE_ARGS="-j 3" export KALDIFEAT_MAKE_ARGS="-j 3"
pip install --verbose kaldifeat pip install --verbose kaldifeat

5
.github/workflows/run-tests-windows-cuda.yml vendored
View File

@ -94,8 +94,9 @@ jobs:
ls -lh * ls -lh *
ls -lh */* ls -lh */*
echo "PWD: $PWD" sed -i '59i#define CUDNN_MAJOR 8' cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive/include/cudnn.h
echo "PWD: $PWD"
- name: Install PyTorch ${{ matrix.torch }} - name: Install PyTorch ${{ matrix.torch }}
shell: bash shell: bash
@ -130,6 +131,7 @@ jobs:
ls -lh ls -lh
- name: Build kaldifeat - name: Build kaldifeat
shell: bash
run: | run: |
cd build_release cd build_release
cmake --build . --target _kaldifeat --config Release cmake --build . --target _kaldifeat --config Release
@ -143,6 +145,7 @@ jobs:
- name: Build wheel - name: Build wheel
shell: bash shell: bash
run: | run: |
export KALDIFEAT_CMAKE_ARGS="-DCMAKE_BUILD_TYPE=$BUILD_TYPE -DCUDNN_INCLUDE_PATH=/d/a/kaldifeat/kaldifeat/cudnn-for-windows/cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive/include -DCUDNN_LIBRARY_PATH=/d/a/kaldifeat/kaldifeat/cudnn-for-windows/cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive/lib/cudnn.lib"
python3 setup.py bdist_wheel python3 setup.py bdist_wheel
ls -lh dist/ ls -lh dist/
pip install ./dist/*.whl pip install ./dist/*.whl