Jetson Orin TorchAudio whl 自编译指南

1
git clone --recursive --branch v2.7.0 https://github.com/pytorch/audio torchaudio
2
cd torchaudio
3
sudo apt-get update && sudo apt-get install -y libsox-dev libimage-exiftool-perl libopus-dev libvpx-dev pkg-config

1
source ../pytorch/.venv/bin/activate

1
export USE_CUDNN=1
2
export CUDNN_INCLUDE_DIR=/usr/include
3
export CUDNN_LIBRARY=/usr/lib/aarch64-linux-gnu/libcudnn.so
4
export USE_CUDA=1
5
export TORCH_CUDA_ARCH_LIST="8.7"
6
export CUDA_HOME=/usr/local/cuda
7
export CUDNN_HOME=/usr/lib/aarch64-linux-gnu
8
export PATH=$CUDA_HOME/bin:$PATH
9
export CPATH="/usr/include/aarch64-linux-gnu:$CPATH"
10
export LIBRARY_PATH="/usr/lib/aarch64-linux-gnu:$LIBRARY_PATH"
11

12
python3 setup.py bdist_wheel

1
uv pip install ./dist/torchaudio-2.7.0a0+654fee8-cp312-cp312-linux_aarch64.whl

1
python -c "
2
import torch; import torchaudio; import torchaudio.transforms as T;
3
print(f'PyTorch version: {torch.__version__}');
4
print(f'TorchAudio version: {torchaudio.__version__}');
5
print(f'Available backends: {torchaudio.list_audio_backends()}');
6
try:
7
    from torchaudio.utils import ffmpeg_utils;
8
    print('FFmpeg build config:', ffmpeg_utils.get_build_config());
9
except ImportError:
10
    print('FFmpeg utils not available in this version.');
11
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu');
12
print(f'Using device: {device}');
13
try:
14
    melspec = T.MelSpectrogram(sample_rate=16000, n_mels=64).to(device);
15
    waveform = torch.randn(1, 16000).to(device);
16
    spec = melspec(waveform);
17
    print('Successfully ran MelSpectrogram on GPU!');
18
    print(f'Output shape: {spec.shape}');
19
except Exception as e:
20
    print(f'GPU operation failed: {e}')
21
"