目录

在 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_64aarch64)编译的预构建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

文档

访问 PyTorch 的全面开发人员文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解答您的问题

查看资源