在 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_64 和 aarch64 (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