Fix installing cuDNN for windows.

This commit is contained in:
Fangjun Kuang 2022-06-03 16:54:32 +08:00
parent db771f4bca
commit 1f4d4c7e43
2 changed files with 7 additions and 2 deletions

View File

@ -94,6 +94,8 @@ jobs:
ls -lh *
ls -lh */*
echo "PWD: $PWD"
- name: Install PyTorch ${{ matrix.torch }}
shell: bash
@ -119,9 +121,12 @@ jobs:
- name: Configure CMake
shell: bash
run: |
echo "PWD: $PWD"
ls -lh
mkdir build_release
cd build_release
cmake -DCMAKE_BUILD_TYPE=$BUILD_TYPE ..
cmake -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 ..
ls -lh
- name: Build kaldifeat

View File

@ -110,7 +110,7 @@ def generate_build_matrix(enable_cuda, for_windows, test_only_latest_torch):
"python-version": ["3.7", "3.8", "3.9", "3.10"],
"cuda": ["10.2", "11.3", "11.5"]
if not for_windows
else ["10.2.89", "11.3.1", "11.5.2"],
else ["11.3.1", "11.5.2"],
},
}
if test_only_latest_torch: