You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何在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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.27 09:35:05