快捷方式

在 Linux 和 macOS 上构建

1. 安装 Conda 并激活 conda 环境

请按照以下说明操作:https://docs.conda.org.cn/en/latest/miniconda.html

2. 安装 PyTorch

请从以下链接选择您要安装的 PyTorch 版本:https://pytorch.ac.cn/get-started/locally/

在这里,我们安装 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 仅支持就地构建。要使用 pip,请使用 --no-use-pep517 选项。

pip install -v -e . --no-use-pep517

[可选] 使用自定义构建的 FFmpeg 构建 TorchAudio

默认情况下,torchaudio 尝试构建 FFmpeg 扩展,以支持多个 FFmpeg 版本。此过程使用为特定 CPU 架构(如 x86_64aarch64 (arm64))编译的预构建 FFmpeg 库。

如果您的 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 的全面开发者文档

查看文档

教程

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

查看教程

资源

查找开发资源并获得问题解答

查看资源