Unable to import name '_TPU_AVAILABLE' from 'pytorch_lightning.utilities'

Question

I'm trying to import aitexten package to work on a GPT-2 solution. But I'm encountering an error:ImportError: cannot import name '_TPU_AVAILABLE' from 'pytorch_lightning.utilities' (/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/init.py)

I tried downgrading pytorch to v1.11.0 but that isn't helping too. Please help!

Intercommunion

Answer 1

This worked for me in Google Colab. And it's with aitextgen version 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 

Scrip

Answer 2

I ran into the same problem! I fixed it by downgrading aitextgen.

In the beginning of my notebook, I ran this command:

!pip3 install -q aitextgen==0.5.2

Make sure to restart your kernel.

Mentalist

Thank you so much for your help! In my case v0.5.1 worked out for aitextgen – Intercommunion

Answer 3

try this :

pytorch-lightning==1.4.0rc0
torchmetrics==0.6.0
aitextgen==0.5.2

Autogamy

Answer 4

In my case

pip install -qq pytorch-lightning==1.7.7 aitextgen==0.6.0

Fixed it.

Firstclass