无法在googlecolab上导入pytorch_lightning
我做了以下工作:
!pip install pytorch_lightning -qqq
import pytorch_lightning
但得到以下错误:
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)
可能是什么问题?
回答
如Github上的 Issue #6415 所述,尝试从 GitHub 安装。
它对我有用。
!pip install git+https://github.com/PyTorchLightning/pytorch-lightning
import pytorch_lightning as pl
print(pl.__version__)
输出:
1.3.0dev
该错误似乎来自问题 #6210,他们说已修复。我猜它没有上传到 PyPi。