dustynv/whisperx

whisperx容器镜像文档

镜像概述和主要用途

whisperx容器镜像是jetson-containers项目的一部分，专为Jetson嵌入式设备设计，提供高效的语音识别与音频处理能力。该镜像基于CUDA加速技术，集成了PyTorch、faster-whisper等深度学习框架及优化工具，适用于需要在边缘设备上实现实时语音转写、音频分析的场景。

核心功能和特性

• GPU加速支持：依赖CUDA、cuDNN实现硬件加速，提升语音处理效率
• 深度学习框架集成：包含PyTorch 2.2、torchaudio等，支持语音模型部署与推理
• 优化工具链：集成faster-whisper、ctranslate2等，优化语音识别性能
• 完整依赖栈：包含Python、numpy、transformers等数据处理与模型库，满足端到端开发需求
• 系统兼容性：要求L4T系统版本≥34.1.0（JetPack 5.1+），兼容Jetson系列设备

容器信息

容器镜像信息

容器兼容性说明：

• L4T R32.7容器可运行于其他L4T R32.7版本（JetPack 4.6+）
• L4T R35.x容器可运行于其他L4T R35.x版本（JetPack 5.1+）

运行容器

使用jetson-containers工具

可通过jetson-containers run结合autotag自动拉取或构建兼容镜像：

bash
# 自动拉取或构建兼容的容器镜像
jetson-containers run $(autotag whisperx)

# 显式指定镜像版本
jetson-containers run dustynv/whisperx:r35.3.1

使用docker run命令

手动指定镜像及运行参数：

bash
sudo docker run --runtime nvidia -it --rm --network=host dustynv/whisperx:r35.3.1

挂载目录与运行命令

挂载主机目录至容器：

bash
jetson-containers run -v /主机路径:/容器路径 $(autotag whisperx)

运行指定命令（非交互式shell）：

bash
jetson-containers run $(autotag whisperx) 应用程序 --参数 数值

说明：jetson-containers run会自动添加默认参数（如--runtime nvidia、挂载/data缓存等），并在执行前打印完整命令。

构建容器

若使用autotag时需要构建镜像，可按以下步骤手动构建：

1. 先完成https://github.com/dusty-nv/jetson-containers/tree/master/docs/setup.md
2. 执行构建命令：

bash
jetson-containers build whisperx

构建过程中会集成依赖组件并进行测试。使用--help查看构建选项：

bash
jetson-containers build whisperx --help