Unable to import pytorch_lightning on google colab

user_15357068 0
Viewed 20270

I have done the following:

!pip install pytorch_lightning -qqq
import pytorch_lightning

But get the following error:

ImportError                               Traceback (most recent call last)
<ipython-input-7-d883b15aac58> in <module>()
----> 1 import pytorch_lightning

----------------------------------9 frames------------------------------------------------
/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/apply_func.py in <module>()
     26 
     27 if _TORCHTEXT_AVAILABLE:
---> 28     from torchtext.data import Batch
     29 else:
     30     Batch = type(None)

ImportError: cannot import name 'Batch' from 'torchtext.data' (/usr/local/lib/python3.7/dist-packages/torchtext/data/__init__.py)

What could the issue be?

pytorch-lightning pytorch google-colaboratory python
7 Answers

As said in Issue #6415 on Github, try installing from the GitHub.
It worked for me.

!pip install git+https://github.com/PyTorchLightning/pytorch-lightning
import pytorch_lightning as pl
print(pl.__version__)

Output:

1.3.0dev

It seems that the error is coming from Issue #6210 and they say it was fixed. I guess it wasn't uploaded to PyPi.

user_142039130
user_44168830

You can try this command, I encountered the same problem and was able to fix the problem.

!pip install --upgrade pytorch-lightning
user_148462150

Late, but if you are having similar difficulties with PyTorch Lightning Bolts you can solve it at the time of posting by making use of the same strategy:

!pip install git+https://github.com/PyTorchLightning/lightning-bolts
user_63569890

Seems like the problem arises from the pytorch-lightning==1.1.x versions. Version above 1.2.x fixes the problem

But taking the latest version as in PythonSnek's answer resulted in some other bugs later on with the checkpoints saving. This could be because the latest version - 1.3.0dev is not still in development.

Installing the tar.gz of one of the stable versions fixes the problem

!pip install https://github.com/PyTorchLightning/pytorch-lightning/releases/download/1.2.6/pytorch-lightning-1.2.6.tar.gz
user_58744760
user_110929100

may be you should upgrade your pytorch-lightning. My version is 1.4.0, when I upgrade it to 1.5.10. The error disappears.

user_193358150
Related