使用pyannote.audio与Whisper时RTX2060 Super GPU未被利用问题求助

问题详情 #

我使用pyannote.audio完成说话人分割，再通过Whisper进行语音转录。两台设备中，搭载RTX3050的笔记本运行代码时GPU占用率达80-100%，但搭载RTX2060 Super的台式机GPU完全未被利用。

两台设备执行以下代码均返回True：

torch.cuda.is_available()

我已通过以下代码强制指定使用GPU：

device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # 返回cuda
model  = whisper.load_model(model_size, device=device)
result = model.transcribe(audio=audiof, language=lang, word_timestamps=True)

设备配置 #

笔记本（GPU正常运行） #

• 已安装Nvidia CUDA Toolkit，版本11.7
• torch相关依赖版本：

torch                     2.0.1+cu117
torch-audiomentations     0.11.0
torch-pitch-shift         1.2.4
torchaudio                2.0.2+cu117
torchmetrics              0.11.4
torchvision               0.15.2+cu117

• Whisper与pyannote相关依赖版本：

openai                    0.27.8
openai-whisper            20230314
pyannote.audio            2.1.1
pyannote.core             5.0.0
pyannote.database         5.0.1
pyannote.metrics          3.2.1
pyannote.pipeline         2.3

台式机（GPU未被利用） #

• 已安装Nvidia CUDA Toolkit，版本11.8
• torch相关依赖版本：

torch                     2.0.1+cu118
torch-audiomentations     0.11.0
torch-pitch-shift         1.2.4
torchaudio                2.0.2+cu118
torchmetrics              0.11.4
torchvision               0.15.2+cu118

• Whisper与pyannote相关依赖版本：

openai                    0.27.8
openai-whisper            20230314
pyannote.audio            2.1.1
pyannote.core             5.0.0
pyannote.database         5.0.1
pyannote.metrics          3.2.1
pyannote.pipeline         2.3

内容的提问来源于stack exchange，提问作者mirzaahmergull