快捷方式

在 Linux 和 macOS 上構建

1. 安裝 Conda 並激活 conda 環境

請按照 https://docs.conda.io/en/latest/miniconda.html 上的說明進行操作

2. 安裝 PyTorch

請從 https://pytorch.com.tw/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 僅支援 in-place build(原地構建)。要使用 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 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並解答您的疑問

檢視資源