Building from source¶
TorchAudio integrates PyTorch for numerical computation and third party libraries for multimedia I/O. It requires the following tools to build from source.
CUDA toolkit and cuDNN (if building CUDA extension)
Customizing the build¶
TorchAudio’s integration with third party libraries can be enabled/disabled via environment variables.
They can be enabled by passing 1
and disabled by 0
.
BUILD_SOX
: Enable/disable I/O features based on libsox.BUILD_KALDI
: Enable/disable feature extraction based on Kaldi.BUILD_RNNT
: Enable/disable custom RNN-T loss function.USE_FFMPEG
: Enable/disable I/O features based on FFmpeg libraries.USE_ROCM
: Enable/disable AMD ROCm support.USE_CUDA
: Enable/disable CUDA support.
For the latest configurations and their default values, please check the source code. https://github.com/pytorch/audio/blob/main/tools/setup_helpers/extension.py