import - 无法从 'rank_zero_only' 导入名称 'pytorch_lightning.utilities.distributed'

Question

我正在使用 VQGAN+CLIP_(Zooming)_(z+quantize_method_with_addons).ipynb Google Repository，当我点击单元格“加载库和定义”时

它发送了一个错误:

ImportError                               Traceback (most recent call last)
<ipython-input-6-fe8fafeed45d> in <module>
     24 from omegaconf import OmegaConf
     25 from PIL import Image
---> 26 from taming.models import cond_transformer, vqgan
     27 import torch
     28 from torch import nn, optim

1 frames
/content/taming-transformers/main.py in <module>
     10 from pytorch_lightning.trainer import Trainer
     11 from pytorch_lightning.callbacks import ModelCheckpoint, Callback, LearningRateMonitor
---> 12 from pytorch_lightning.utilities.distributed import rank_zero_only
     13 
     14 from taming.data.utils import custom_collate

ImportError: cannot import name 'rank_zero_only' from 'pytorch_lightning.utilities.distributed' (/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/distributed.py)

我不知道如何解决这个问题。我不知道如何手动安装 Pytorch，因为它说“注意:如果您的导入由于缺少包而失败，您可以 使用 !pip 或 !apt 手动安装依赖项。

要查看安装一些常见依赖项的示例，请单击 下面的“打开示例”按钮。”

如果您有解决方案，请提前致谢。

伊内丝

我 triad !pip install 但我可能真的不知道把这个单元格/代码行放在哪里

Answer 1

pytorch_lightning 最近发布了一个会抛出此错误的新版本(2022 年 11 月 2 日发布的版本 1.8.0.post1)。

https://pypi.org/project/pytorch-lightning/#history

只需安装旧版本的 pytorch_lightning 即可。

在我的系统中，我运行了“pip install pytorch-lightning==1.6.5”，更高版本也可以，您可以通过点击上面提供的链接然后点击发布历史来查看它们。