在 pytorch 闪电中导入错误

Question

我已经使用以下命令安装了

pytorch_lightning

和

pl_bolts

：

1. !pip install pytorch-lightning --upgrade

2. !pip install pytorch-lightning-bolts

但是，当我导入

pl_bolts

时，出现以下错误：

ImportError Traceback (most recent call last)
in <cell line: 1>()
----> 1 import pl_bolts

2 frames
/usr/local/lib/python3.10/dist-packages/pl_bolts/callbacks/data_monitor.py in
4 import torch
5 from pytorch_lightning import Callback, LightningModule, Trainer
----> 6 from pytorch_lightning.loggers import LightningLoggerBase, TensorBoardLogger, WandbLogger
7 from pytorch_lightning.utilities import rank_zero_warn
8 from pytorch_lightning.utilities.apply_func import apply_to_collection

ImportError: cannot import name 'LightningLoggerBase' from 'pytorch_lightning.loggers' (/usr/local/lib/python3.10/dist-packages/pytorch_lightning/loggers/init.py)

我当前环境有以下包版本：

• Python版本：3.10.11
• 火炬版本：2.0.0+cu118
• 火炬视觉版本：0.15.1+cu118
• Pytorch-闪电版：2.0.2
• Pytorch-lightning-bolts 版本：0.6.0.post1

导入没有按预期工作。

Answer 1

不幸的是，您目前需要使用 PyTorch Lightning 1.9 才能使用 Lightning Bolts。如果您想通过将 Bolts 升级到最新的 Lightning 版本来做出贡献，请随时联系 Discord。