无法在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。


以上是无法在googlecolab上导入pytorch_lightning的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>