whisperx模型版本兼容问题及旧版依赖安装失败求助

问题概述 #

运行WhisperX时触发版本不兼容警告：当前使用pyannote.audio 3.1.1、Torch 2.3.1+cpu，与模型训练依赖的pyannote.audio 0.0.1、Torch 1.10.0+cu102不匹配。尝试安装pyannote.audio 0.0.1时，因无法找到满足pytorch-lightning<1.7,>=1.5.4的适配版本而失败。

解决方案 #

方案1：创建兼容虚拟环境（推荐） #

旧版pyannote.audio依赖的pytorch-lightning版本仅支持Python 3.8~3.9（更高Python版本不再提供对应包），需通过虚拟环境隔离依赖：

1. 创建并激活指定Python版本的虚拟环境（以Python 3.9为例）：

# Conda方式
conda create -n whisperx_env python=3.9
conda activate whisperx_env

# Venv方式
python3.9 -m venv whisperx_env
source whisperx_env/bin/activate  # Linux/macOS
whisperx_env\Scripts\activate     # Windows

2. 安装匹配训练环境的Torch版本：

pip install torch==1.10.0+cu102 torchvision==0.11.1+cu102 torchaudio==0.10.0+cu102 -f https://download.pytorch.org/whl/cu102/torch_stable.html

3. 手动安装兼容的pytorch-lightning版本：

pip install pytorch-lightning==1.6.5

4. 安装pyannote.audio 0.0.1：

pip install pyannote.audio==0.0.1

5. 安装WhisperX：

pip install whisperx

方案2：忽略版本警告继续使用 #

若实际运行未出现功能异常，可直接屏蔽警告（警告仅提示风险，部分场景下新版本仍能兼容旧模型）：

import warnings
warnings.filterwarnings("ignore", category=UserWarning, message="The model was trained with pyannote.audio.*")
warnings.filterwarnings("ignore", category=UserWarning, message="The model was trained with torch.*")

方案3：升级WhisperX适配新版本依赖 #

检查WhisperX最新版本是否已适配pyannote.audio 3.x，尝试升级：

pip install --upgrade whisperx

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