無法在 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__)

Output：

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 閃電未引用最新的火炬文本。

Answer 3

你可以試試這個命令，我遇到了同樣的問題並且能夠解決這個問題。

!pip install --upgrade pytorch-lightning

Answer 4

遲到了，但是如果您在使用 PyTorch 閃電螺栓時遇到類似的困難，您可以在發布時使用相同的策略解決它：

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

Answer 5

似乎問題來自pytorch-lightning==1.1.x版本。 1.2.x以上的版本修復了這個問題

但是，在PythonSnek的答案中采用最新版本會導致稍后在保存檢查點時出現其他一些錯誤。 這可能是因為最新版本 - 1.3.0dev尚未開發。

安裝其中一個 穩定版本的tar.gz可以解決問題

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

Answer 6

進行以下安裝解決了我的問題

pip install pytorch-lightning==1.4.4

pip install omegaconf -U

pip install hydra-core --upgrade

解決方案來源： https://github.com/PyTorchLightning/pytorch-lightning/issues/7110

Answer 7

可能你應該升級你的 pytorch-lightning。 我的版本是 1.4.0，當我升級到 1.5.10 時。 錯誤消失。