在 Linux 和 macOS 上构建¶
1. 安装 Conda 并激活 conda 环境¶
2. 安装 PyTorch¶
请从 https://pytorch.org/get-started/locally/ 选择要安装的 PyTorch 版本
在这里,我们安装 nightly build。
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 仅支持就地构建。
要使用 ,请使用 option。pip
--no-use-pep517
pip install -v -e . --no-use-pep517
[可选]使用自定义构建的 FFmpeg 构建 TorchAudio¶
默认情况下,torchaudio 会尝试构建支持多个 FFmpeg 版本的 FFmpeg 扩展。此过程使用为特定 CPU 架构(如 和 )编译的预构建 FFmpeg 库。x86_64
aarch64
arm64
如果您的 CPU 不是其中之一,则构建过程可能会失败。要解决此问题,可以禁用 FFmpeg 集成(通过设置环境变量)或切换到单个版本的 FFmpeg 扩展。USE_FFMPEG=0
要构建单个版本的 FFmpeg 扩展,FFmpeg 二进制文件必须由用户提供,并在构建环境中可用。为此,请安装 FFmpeg 并设置环境变量以指定 FFmpeg 的位置。FFMPEG_ROOT
conda install -c conda-forge ffmpeg
FFMPEG_ROOT=${CONDA_PREFIX} python setup.py develop