python - colab中的torchtext ImportError

Question

我正在尝试运行 this tutorial在合作实验室。
但是，当我尝试导入一堆模块时:

import io
import torch
from torchtext.utils import download_from_url
from torchtext.data.utils import get_tokenizer
from torchtext.vocab import build_vocab_from_iterator

它给了我 extract_archive的错误和 build_vocab_from_iterator :

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-5-a24e72502dbc> in <module>()
      1 import io
      2 import torch
----> 3 from torchtext.utils import download_from_url, extract_archive
      4 from torchtext.data.utils import get_tokenizer
      5 from torchtext.vocab import build_vocab_from_iterator

ImportError: cannot import name 'extract_archive'


ImportError                               Traceback (most recent call last)
<ipython-input-4-02a401fd241b> in <module>()
      3 from torchtext.utils import download_from_url
      4 from torchtext.data.utils import get_tokenizer
----> 5 from torchtext.vocab import build_vocab_from_iterator
      6 
      7 url = 'https://s3.amazonaws.com/research.metamind.io/wikitext/wikitext-2-v1.zip'

ImportError: cannot import name 'build_vocab_from_iterator'

请帮我解决这个问题。

Answer 1

你需要先升级torchtext

!pip install -U torchtext==0.8.0

目前，0.8.0 版本可与torch 1.7.0 一起使用(无需升级torch、torchvision)
更新(2021 年 9 月)
目前，torchtext 已经是 0.10.0，你不需要升级任何东西。