import - 无法从 '_TPU_AVAILABLE' 导入名称 'pytorch_lightning.utilities'

Question

我正在尝试导入 aitexten 包以处理 GPT-2 解决方案。但是我遇到了一个错误:ImportError: cannot import name '_TPU_AVAILABLE' from 'pytorch_lightning.utilities' (/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/init .py)

我尝试将 pytorch 降级到 v1.11.0，但这也无济于事。请帮忙!

Answer 1

这在 Google Colab 中对我有用。它与 aitextgen 版本 0.6 一起使用。

!pip uninstall -qqy torch torchvision torchtext torchaudio fastai
!pip install -qq torch==1.9.0 pytorch-lightning==1.7.7 aitextgen==0.6.0