Google cloud platform 如何在google colab TPU上使用torchaudio和torch_xla？

我使用谷歌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版本之间的冲突

以下是我用来安装Libraries的代码：

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

此处提出的解决方案产生以下错误：

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

---

我做错了什么？

运行安装脚本（）后，能否尝试从源代码安装

torchaudio

？听起来好像是因为版本不匹配（夜间火炬与稳定火炬）。

它正在工作！！从源代码安装

torchaudio

修复了问题，我还删除了以下行：

！pip安装torchvision==0.6.1！pip安装火炬==1.5.1

！pip安装torchaudio==0.5.1