从源代码构建

TorchAudio 集成了 PyTorch 进行数值计算。它需要以下工具来从源代码构建。

• PyTorch

• CMake

• Ninja

• 支持 C++ 17 的 C++ 编译器

  • GCC (Linux)

  • Clang (macOS)

  • MSVC 2019 或更新版本 (Windows)

• CUDA toolkit 和 cuDNN (如果构建 CUDA 扩展)

• 在 Linux 和 macOS 上构建
• 在 Windows 上构建
• 在 Jetson 上构建

自定义构建

可以通过环境变量启用/禁用 torchaudio 的可选功能。

可以通过传递 1 来启用它们，通过传递 0 来禁用它们。

• BUILD_RNNT: 启用/禁用自定义 RNN-T 损失函数。

• USE_ROCM: 启用/禁用 AMD ROCm 支持。

• USE_CUDA: 启用/禁用 CUDA 支持。

有关最新的配置及其默认值，请检查源代码。 https://github.com/pytorch/audio/blob/main/tools/setup_helpers/extension.py