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

Print TORCH_DIR for debug. (#44)

Browse Source 
* Print TORCH_DIR for debug.

* fix style issues

* test torch 1.12.0
This commit is contained in:
Fangjun Kuang 2022-07-20 10:08:47 +08:00 committed by GitHub
parent e15b2ae6f5
commit e78f081327
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 33 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.flake8
View File

@ -3,6 +3,7 @@ max-line-length = 80
exclude = exclude =
.git, .git,
doc,
build, build,
build_release, build_release,
cmake/cmake_extension.py, cmake/cmake_extension.py,

1
cmake/torch.cmake
View File

@ -8,6 +8,7 @@ execute_process(
OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_STRIP_TRAILING_WHITESPACE
OUTPUT_VARIABLE TORCH_DIR OUTPUT_VARIABLE TORCH_DIR
) )
message(STATUS "TORCH_DIR: ${TORCH_DIR}")
list(APPEND CMAKE_PREFIX_PATH "${TORCH_DIR}") list(APPEND CMAKE_PREFIX_PATH "${TORCH_DIR}")
find_package(Torch REQUIRED) find_package(Torch REQUIRED)

8
scripts/github_actions/generate_build_matrix.py
View File

@ -112,9 +112,15 @@ def generate_build_matrix(enable_cuda, for_windows, test_only_latest_torch):
if not for_windows if not for_windows
else ["11.3.1", "11.5.2"], else ["11.3.1", "11.5.2"],
}, },
"1.12.0": {
"python-version": ["3.7", "3.8", "3.9", "3.10"],
"cuda": ["10.2", "11.3", "11.6"]
if not for_windows
else ["11.3.1", "11.6.2"],
},
} }
if test_only_latest_torch: if test_only_latest_torch:
latest = "1.11.0" latest = "1.12.0"
matrix = {latest: matrix[latest]} matrix = {latest: matrix[latest]}
ans = [] ans = []

3
scripts/github_actions/install_cuda.sh
View File

@ -43,6 +43,9 @@ case "$cuda" in
11.5) 11.5)
url=https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux.run url=https://developer.download.nvidia.com/compute/cuda/11.5.2/local_installers/cuda_11.5.2_495.29.05_linux.run
;; ;;
11.6)
url=https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
;;
*) *)
echo "Unknown cuda version: $cuda" echo "Unknown cuda version: $cuda"
exit 1 exit 1

8
scripts/github_actions/install_cudnn.sh
View File

@ -36,9 +36,9 @@ case $cuda in
11.5) 11.5)
filename=cudnn-11.3-linux-x64-v8.2.0.53.tgz filename=cudnn-11.3-linux-x64-v8.2.0.53.tgz
;; ;;
# 11.5) 11.6)
# filename=cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz filename=cudnn-11.3-linux-x64-v8.2.0.53.tgz
# ;; ;;
*) *)
echo "Unsupported cuda version: $cuda" echo "Unsupported cuda version: $cuda"
exit 1 exit 1
@ -47,7 +47,7 @@ esac
command -v git-lfs >/dev/null 2>&1 || { echo >&2 "\nPlease install 'git-lfs' first."; exit 2; } command -v git-lfs >/dev/null 2>&1 || { echo >&2 "\nPlease install 'git-lfs' first."; exit 2; }
git clone https://huggingface.co/csukuangfj/cudnn GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/csukuangfj/cudnn
cd cudnn cd cudnn
git lfs pull --include="$filename" git lfs pull --include="$filename"

17
scripts/github_actions/install_torch.sh
View File

@ -125,6 +125,23 @@ case ${torch} in
;; ;;
esac esac
;; ;;
1.12.*)
case ${cuda} in
10.2)
package="torch==${torch}"
# Leave it empty to use PyPI.
url=
;;
11.3)
package="torch==${torch}+cu113"
url=https://download.pytorch.org/whl/torch_stable.html
;;
11.6)
package="torch==${torch}+cu116"
url=https://download.pytorch.org/whl/torch_stable.html
;;
esac
;;
*) *)
echo "Unsupported PyTorch version: ${torch}" echo "Unsupported PyTorch version: ${torch}"
exit 1 exit 1