Как использовать torchaudio с torch_xla в google colab TPU?

Я использовал Google Colab (с включенным графическим процессором) для обучения моей модели автоматического распознавания речи на основе 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

Что я делаю неправильно ?


pytorch google-cloud-platform torchvision tpu
person MKDYasserH    schedule 09.07.2020    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Можете ли вы попробовать установить torchaudio из исходного кода после запуска сценария установки (https://github.com/pytorch/audio#from-source)? Похоже, это из-за несоответствия версий (ночной фонарик против стабильного torchaudio).

person jysohn    schedule 14.07.2020
comment
Работает !! Установка torchaudio из исходного кода устранила проблему. Я также удалил следующие строки: !pip install torchvision==0.6.1 !pip install torch==1.5.1 !pip install torchaudio==0.5.1 - person MKDYasserH; 15.07.2020