无法从“pytorch_lightning.utilities”导入名称“_TPU_AVAILABLE”

Question

我正在尝试导入 aitexten 包以在 GPT-2 解决方案上工作。但我遇到了一个错误：ImportError：无法从“pytorch_lightning.utilities”导入名称“_TPU_AVAILABLE”（/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/ init .py）

我尝试将 pytorch 降级到 v1.11.0 但这也没有帮助。请帮忙！

Answer 1

我遇到了同样的问题！我通过降级修复了它aitextgen。

在我的笔记本的开头，我运行了以下命令：

!pip3 install -q aitextgen==0.5.2

确保重新启动内核。