如何在谷歌 colab tpu 上使用 torchaudio 和 torch xla答案

【问题标题】：how to use torchaudio with torch xla on google colab tpu如何在谷歌 colab tpu 上使用 torchaudio 和 torch xla
【发布时间】：2020-03-17 08:39:06
【问题描述】：

我正在尝试运行一个在谷歌 TPU 上使用 torchaudio 的 pytorch 脚本。为此，我在this notebook 之后使用pytorch xla，更具体地说，我使用此代码单元来加载xla：

!pip install torchaudio
import os
assert os.environ['COLAB_TPU_ADDR'], 'Make sure to select TPU from Edit > Notebook settings > Hardware accelerator'

VERSION = "20200220"  #@param ["20200220","nightly", "xrt==1.15.0"]
!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
!python pytorch-xla-env-setup.py --version $VERSION

import torch

import torchaudio

import torch_xla

但是这与我需要的 torchaudio 版本不兼容：ERROR: torchaudio 0.4.0 has requirement torch==1.4.0, but you'll have torch 1.5.0a0+e95282a which is incompatible.

我在任何地方都找不到如何使用 pytorch xla 加载 torch 1.4.0。

我尝试使用夜间版本的手电筒音频，但错误如下：

!pip install torchaudio_nightly -f https://download.pytorch.org/whl/nightly/torch_nightly.html

import os
assert os.environ['COLAB_TPU_ADDR'], 'Make sure to select TPU from Edit > Notebook settings > Hardware accelerator'

VERSION = "20200220"  #@param ["20200220","nightly", "xrt==1.15.0"]
!curl https://raw.githubusercontent.com/pytorch/xla/master/contrib/scripts/env-setup.py -o pytorch-xla-env-setup.py
!python pytorch-xla-env-setup.py --version $VERSION

import torch
import torchaudio

import torch_xla

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-2-968e9d93c06f> in <module>()
      9 
     10 import torch
---> 11 import torchaudio
     12 
     13 import torch_xla

/usr/local/lib/python3.6/dist-packages/torchaudio/__init__.py in <module>()
      3 
      4 import torch
----> 5 import _torch_sox
      6 
      7 from .version import __version__, git_version

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

---------------------------------------------------------------------------

那么我将如何使用 xla 加载 pytorch 的稳定版本或 1.4.0 版本，或者对于这种情况还有其他解决方法吗？

非常感谢您的帮助！

【问题讨论】：

你好。您创建了哪种类型的 TPU？还有什么 TPU 软件版本？
笔记本链接过时，出现 404。我用这个笔记本测试过。colab.sandbox.google.com/github/pytorch/xla/blob/master/contrib/… 如果这对你不起作用，我会在下面发布一个答案让我知道，我会继续测试一些更多。也许你也可以查看笔记本的链接。当我第一次使用上面的笔记本时，我也收到了同样的错误“错误：torchaudio 0.4.0 需要torch==1.4.0，但你将拥有torch 1.5。 0a0+e95282a 不兼容。但做了一些更改，笔记本中的所有单元格都已成功完成

标签： google-cloud-platform pytorch tpu

【解决方案1】：

我使用下面的笔记本进行了测试； Getting Started with PyTorch on Cloud TPUs

更改单元格后包含；

来自：

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

到：

    VERSION = "20200325"  #@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
    !pip install torch==1.4.0
    !pip install torchaudio==0.4.0
    %matplotlib inline
    !python pytorch-xla-env-setup.py --version $VERSION

所有单元格都运行成功，下面的导入语句没有报错；

    # imports pytorch
      import torch

    # imports the torch_xla package
      import torch_xla
      import torch_xla.core.xla_model as xm

【讨论】：

【解决方案2】：

!pip uninstall torch -y
!pip install torch==1.6

如果看起来像..“警告：root:TPU 已成功启动，版本为 pytorch-1.6”

享受...！！

【讨论】：