无法在 google colab 上导入 pytorch_lightning

Question

我做了以下工作：

!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)

可能是什么问题？

Answer 1

如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。

Answer 2

示例工作环境：https://colab.research.google.com/drive/1GSCd3Gz3EOQIln3v065VKWKbB3_F8xqK ?usp=sharing

您可以在重新启动环境后尝试吗？

!pip install torchtext==0.8.0 torch==1.7.1 pytorch-lightning==1.2.2

import pytorch_lightning as pl
print(pl.__version__)
...

似乎有一个错误尚未影响 pip，因为 pytorch lighting 未引用最新的 torchtext。