求适用于NVIDIA GPU平台aarch64的匹配版torchaudio预编译wheel包
问题描述
NVIDIA提供了对应JetPack 5.0.2版本的GPU加速PyTorch预编译wheel包,但未包含torchaudio组件。经测试必须使用与该PyTorch严格匹配的torchaudio版本,否则即便版本号一致,也会触发如下错误:
OSError: ../lib/libtorchaudio.so: undefined symbol: _ZNK5torch8autograd4Node4nameEv
请问该匹配版本的torchaudio预编译wheel包可从何处获取?不考虑使用NVIDIA提供的JetPack Docker镜像(小型AI计算机上不适用)。
解决方案
- 优先从NVIDIA官方的L4T(Linux for Tegra)配套资源中获取对应JetPack版本的torchaudio预编译wheel包,这类包是和NVIDIA发布的PyTorch版本严格适配的,可直接避免符号缺失问题。
- 若官方下载页未直接列出该包,可查阅NVIDIA发布的PyTorch版本说明文档,找到对应JetPack版本匹配的torchaudio版本及获取渠道(需确保是针对ARM64架构+CUDA的版本)。
- 若无法找到预编译包,可手动编译适配版本:
- 确认当前环境中PyTorch的具体版本和CUDA版本(JetPack 5.0.2对应CUDA 11.4)
- 下载与该PyTorch版本完全匹配的torchaudio源码(通过git checkout对应版本标签)
- 设置环境变量
USE_CUDA=1,执行pip install .完成源码编译安装,或执行python setup.py bdist_wheel生成wheel包后再安装
内容的提问来源于stack exchange,提问作者k_o_
相关产品推荐
相关产品推荐

