如何在 google colab TPU 上使用 torchaudio 和 torch_xla？答案

【问题标题】：How do I use torchaudio with torch_xla on google colab TPU?如何在 google colab TPU 上使用 torchaudio 和 torch_xla？
【发布时间】：2020-10-30 01:27:19
【问题描述】：

我正在使用 google colab（启用 GPU）来训练我的基于 pytorch 和 torchaudio 的自动语音识别模型。但是当我尝试使用 google colab TPU 时，我在训练我的模型时遇到了以下错误：

ImportError: /usr/local/lib/python3.6/dist-packages/_torch_sox.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN6caffe28TypeMeta21_typeMetaDataInstanceISt7complexIfEEEPKNS_6detail12TypeMetaDataEv site:stackoverflow.com

我几乎可以肯定它是 torchaudio 和 pytorch_xla 版本之间的冲突

这是我用来安装库的代码：

VERSION = "nightly"  #@param ["1.5" , "20200325", "nightly"]
!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
!pip install torchvision==0.6.1
!pip install torch==1.5.1
!pip install torchaudio==0.5.1
!python pytorch-xla-env-setup.py --version $VERSION

这里提出的解决方案https://stackoverflow.com/a/60929133/13847989 产生了以下错误：

ImportError: /usr/local/lib/python3.6/dist-packages/_torch_sox.cpython-36m-x86_64-linux-gnu.so: undefined symbol: THPVariableClass

我做错了什么？

【问题讨论】：

标签： google-cloud-platform pytorch tpu torchvision

【解决方案1】：

您可以在运行安装脚本 (https://github.com/pytorch/audio#from-source) 后尝试从源代码安装 torchaudio 吗？听起来像是由于版本不匹配（夜间火炬与稳定的火炬音频）。

【讨论】：

它的工作！从源安装torchaudio 解决了这个问题我还删除了这些行：!pip install torchvision==0.6.1!pip install torch==1.5.1!pip install torchaudio==0.5.1