如何在CPU上运行pyannote.audio的speaker-diarization流水线?
将pyannote Speaker Diarization切换到CPU运行的解决方案
运行pyannote的speaker-diarization流水线时出现CUDA内存不足错误,只需通过指定运行设备为CPU即可解决,以下是两种可行的修改方式:
方式一:加载流水线后指定设备
from pyannote.audio import Pipeline pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token="token...") # 强制设置流水线使用CPU pipeline.device = "cpu" diarization = pipeline("example.wav")
方式二:调用流水线时传入设备参数
from pyannote.audio import Pipeline pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization", use_auth_token="token...") # 调用流水线时指定使用CPU diarization = pipeline("example.wav", device="cpu")
两种方式都能让speaker-diarization流水线完全在CPU上运行,规避CUDA内存不足的问题。
内容的提问来源于stack exchange,提问作者user3668129
相关产品推荐
相关产品推荐

