Docker 容器化 Whisper：镜像选型与模型预加载实战
Whisper 是 OpenAI 开源的语音识别模型，支持多语言转录和翻译。通过 Docker 容器化部署，不仅能隔离环境，还能提升可移植性和资源管理效率。实际使用中，大家常遇到的痛点是首次运行时的模型下载延迟——大模型动辄几 GB，跑起来慢得让人着急。这篇内容结合实战经验，聊聊如何选择合适的镜像版本，并在构建阶段预下载模型文件，让容器启动即服务。
1. 了解模型版本差异
Whisper 提供了不同参数量级的模型，选择时需权衡精度与资源消耗：
- tiny / base：体积极小，适合低配设备或快速测试。
- small / medium：平衡性较好，medium 推荐用于大多数生产场景。
- large：精度最高，但显存和内存占用较大。
模型文件首次加载时会自动下载并缓存在 ~/.cache/whisper。在容器内，这个路径默认位于 /root/.cache/whisper。如果每次启动都重新下载，体验会很差，所以最佳实践是在构建镜像时就把它塞进去。
2. 镜像来源与版本选择
OpenAI 官方并未提供直接的 Docker 镜像，但社区维护得非常完善。推荐使用 GitHub Container Registry (GHCR) 上的镜像，稳定性不错。
- 基础环境：建议基于
python:3.11-slim，轻量且干净。 - 完整镜像：如
ghcr.io/openai/whisper:latest，通常已包含依赖。 - 标签策略：虽然有些镜像支持按模型大小打标签（如
whisper-medium），但更灵活的做法是统一使用基础镜像，通过环境变量控制加载哪个模型。
测试环境可以用 tiny 跑通流程，生产环境则根据硬件情况锁定 medium 或 large。记得用 docker pull <image>:<tag> 确认可用版本后再操作。
3. 构建时预下载模型
核心思路是利用 Dockerfile 的 RUN 指令，在镜像构建阶段触发 Python 脚本加载模型，将缓存文件固化到镜像层中。这样容器启动时就不需要再联网下载了。
完整 Dockerfile 示例
下面是一个构建 medium 模型镜像的参考方案。你可以根据需求修改 MODEL_NAME 变量。
# 使用轻量级 Python 基础镜像
FROM python:3.11-slim
# 设置环境变量，指定要下载的模型版本
ENV MODEL_NAME=medium
# 安装 Whisper 及音频处理依赖
RUN pip install --no-cache-dir openai-whisper ffmpeg-python
# 创建临时脚本，用于触发模型下载
RUN echo "import whisper; whisper.load_model('${MODEL_NAME}')" > /app/download_model.py
# 执行脚本，将模型缓存写入镜像
RUN python /app/download_model.py
# 清理临时脚本，减小镜像体积
RUN rm /app/download_model.py
# 设置工作目录
WORKDIR /app
# 默认命令，实际使用时可通过 docker run 覆盖
CMD ["whisper", "--help"]
这里的关键点在于第 10-12 行。我们生成一个 Python 脚本调用 load_model，这会强制 Whisper 把模型文件下载到缓存目录。因为这是在构建阶段运行的，所以模型文件会永久保存在镜像里。最后一步删除脚本是为了避免浪费空间。
构建与运行
在终端执行以下命令构建镜像：
docker build -t whisper-medium:latest .
构建完成后，直接运行容器进行转录测试：
docker run -it -- whisper-medium:latest whisper --model medium --task transcribe --language en audio.wav
