在 Linux 和 macOS 上构建¶
1. 安装 Conda 并激活 conda 环境¶
请按照 https://docs.conda.io/en/latest/miniconda.html 中的说明进行操作
2. 安装 PyTorch¶
请选择您要安装的PyTorch版本,从 https://pytorch.org/get-started/locally/
此处,我们安装夜间构建版本。
conda install pytorch -c pytorch-nightly
3. 安装构建工具¶
conda install cmake ninja
4. 克隆 torchaudio 仓库¶
git clone https://github.com/pytorch/audio
cd audio
5. 构建¶
python setup.py develop
注意
由于构建过程的复杂性,TorchAudio 仅支持原地构建。
要使用 pip,请使用 --no-use-pep517 选项。
pip install -v -e . --no-use-pep517
[可选] 使用自定义构建的 FFmpeg 构建 TorchAudio¶
默认情况下,torchaudio会尝试构建支持多种FFmpeg版本的FFmpeg扩展。此过程使用为特定CPU架构(如 x86_64 和 aarch64)编译的预构建FFmpeg库 (arm64)。
如果您的CPU不属于上述情况,那么构建过程可能会失败。为解决此问题,可以禁用FFmpeg集成(通过设置环境变量USE_FFMPEG=0)或切换到单版本FFmpeg扩展。
要构建单版本FFmpeg扩展,用户必须提供FFmpeg二进制文件,并确保其在构建环境中可用。为此,请安装FFmpeg,并将FFMPEG_ROOT环境变量设置为指定FFmpeg的位置。
conda install -c conda-forge ffmpeg
FFMPEG_ROOT=${CONDA_PREFIX} python setup.py develop