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 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" 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 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