AWS SageMaker安装openai-whisper失败求助：进程终止且模块缺失

问题根源 #

安装进程在Collecting torch阶段被Killed，核心原因是SageMaker notebook实例内存不足——torch包体积较大，源码构建或下载安装过程会占用大量内存，默认小内存实例（如t2.medium）无法支撑。

解决步骤 #

1. 升级实例类型（最直接有效） #

• 停止当前SageMaker notebook实例
• 在实例设置中，将实例类型更换为t3.large（8GB内存）或更高配置（如t3.xlarge）
• 重启实例后重新执行安装命令

2. 优化安装命令，避免内存过载 #

先安装预编译的CPU版torch（无需源码编译，减少内存占用），再安装whisper：

# 安装预编译CPU版torch
pip install torch==2.0.0+cpu torchvision==0.15.1+cpu torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cpu

# 安装openai-whisper，禁用缓存减少内存占用
pip install openai-whisper --no-cache-dir

3. 使用SageMaker自带conda环境 #

SageMaker自带的conda环境配置更稳定，避免虚拟环境的内存冲突：

# 激活Python3.9的conda环境
conda activate python39

# 执行上述torch和whisper安装命令

4. 清理内存后重试 #

如果暂时无法升级实例，先释放现有内存：

# 关闭闲置的ipykernel进程
pkill -f ipykernel

# 清理pip缓存和系统缓存
pip cache purge
sudo apt clean

清理完成后再执行安装命令。

验证安装 #

安装完成后，运行以下代码验证：

import whisper
# 加载基础模型测试
model = whisper.load_model("base")

无报错则安装成功。

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