pytorch 如何在CPU上运行pyannote.audio？

agxfikkp 于 2023-10-20 发布在其他

关注(0)|答案(1)|浏览(145)

我正在运行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上运行扬声器日志化管道？

pytorch

来源：https://stackoverflow.com/questions/75820986/how-to-run-pyannote-audio-on-cpu

1条答案

按热度按时间

eeq64g8w1#

你可以用手电筒

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

赞(0）回复(0）举报 2023-10-20

我来回答

pytorch 如何在CPU上运行pyannote.audio？

1条答案

相关问题

热门标签

最新问答