From fc2beefaebfb4e70a50c48d0cb23766cf58cf84d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20BREDIN?= Date: Wed, 27 Sep 2023 12:34:57 +0200 Subject: [PATCH 1/4] wip: switch to onnxruntime-gpu --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 888ada78a..7829ada37 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ einops >=0.6.0 huggingface_hub >= 0.13.0 lightning >= 2.0.1 omegaconf >=2.1,<3.0 -onnxruntime >= 1.16.0 +onnxruntime-gpu >= 1.16.0 pyannote.core >= 5.0.0 pyannote.database >= 5.0.1 pyannote.metrics >= 3.2 From f7a041b8ca1c8ce9dd1928f1f28808c55f995604 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20BREDIN?= Date: Wed, 27 Sep 2023 14:50:58 +0200 Subject: [PATCH 2/4] wip: use default search algorithm instead of benchmarking --- pyannote/audio/pipelines/speaker_verification.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/pyannote/audio/pipelines/speaker_verification.py b/pyannote/audio/pipelines/speaker_verification.py index 594a4823c..6b39679dd 100644 --- a/pyannote/audio/pipelines/speaker_verification.py +++ b/pyannote/audio/pipelines/speaker_verification.py @@ -447,7 +447,14 @@ def to(self, device: torch.device): if device.type == "cpu": providers = ["CPUExecutionProvider"] elif device.type == "cuda": - providers = ["CUDAExecutionProvider"] + providers = [ + ( + "CUDAExecutionProvider", + { + "cudnn_conv_algo_search": "DEFAULT", # EXHAUSTIVE / HEURISTIC / DEFAULT + }, + ) + ] else: warnings.warn( f"Unsupported device type: {device.type}, falling back to CPU" From 73d80636086a4c4d4bd20621fd209ced96c02d31 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20BREDIN?= Date: Thu, 28 Sep 2023 21:34:15 +0200 Subject: [PATCH 3/4] doc: update changelog --- CHANGELOG.md | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index c444380d7..d63c193d0 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,9 @@ # Changelog +## Version 3.0.1 (2023-09-28) + + - fix(pipeline): fix WeSpeaker GPU support + ## Version 3.0.0 (2023-09-26) ### Features and improvements From 36d224af38c34e82a6f5e0dd08e3f4684bca0402 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Herv=C3=A9=20BREDIN?= Date: Thu, 28 Sep 2023 21:35:06 +0200 Subject: [PATCH 4/4] ci: bump version --- version.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/version.txt b/version.txt index 4a36342fc..cb2b00e4f 100644 --- a/version.txt +++ b/version.txt @@ -1 +1 @@ -3.0.0 +3.0.1