在 AWS 培訓容器上安裝 torchaudio

Question

在 AWS sagemaker GPU 容器中安裝 torchaudio 的正確方法是什么？ 為什么沒有預先安裝？

我試過pip install torchaudio但它也嘗試卸載和安裝 pytorch 以及破壞整個容器。 我收到segmentation fault錯誤可能是因為它安裝了基於 CPU 的版本 pytorch 和 torchaudio。

Answer 1

pip 自動卸載並安裝 Pytorch 的事實可能表明您正在安裝與pytorch 版本不兼容的 torchaudio 版本。 也許：

1. 找到一組匹配的 PyTorch 和 TorchAudio。 測試它們是否在本地安裝良好。
2. 確保找到一個PyTorch 版本的 AWS DLC 容器。
3. pip 安裝 torchaudio 指定您需要的確切版本。 在本地機器上用 Docker 測試這個，不需要雲。