如何在CPU上运行pyannote.audio？

Question

我正在尝试运行

pyannote

并收到错误：“

CUDA out of memory

”。

因此，我想在 CPU 而不是 GPU 上运行我的代码。

from pyannote.audio import Pipeline

pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization",use_auth_token="token...")

diarization = pipeline("example.wav")

在给定上述配置的情况下，如何在 CPU 上运行扬声器二值化管道？

Answer 1

你可以使用手电筒

from pyannote.audio import Pipeline
import torch

pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization",use_auth_token="token...")
device = torch.device("cpu")
pipeline.to(device)
diarization = pipeline("example.wav")