Question

我当时正在使用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

我在做什么错了？

Answer 1

运行设置脚本（https://github.com/pytorch/audio#from-source）之后，您可以尝试从源代码安装torchaudio吗？听起来像是由于版本不匹配（每晚的手电筒与稳定的手电筒音频）。

如何在Google Colab TPU上将torchaudio与torch_xla结合使用？

1 个答案: