Google Colab使用PyTorch Lightning报无LightningModule属性错误如何解决

报错信息：AttributeError: module 'pytorch_lightning' has no attribute 'LightningModule'

原因与对应解决方法 #

• 包版本不兼容
  PyTorch Lightning 2.0版本之后官方将核心包迁移到了lightning名下，旧的pytorch-lightning包和新包存在导入路径差异。首先执行命令检查已安装的包版本：
  !pip list | grep lightning
  • 若已安装的是2.0+版本的lightning包，将导入语句修改为以下形式即可：

    from lightning.pytorch import LightningModule, Trainer
    

  • 若要沿用旧版pytorch-lightning的导入语法，先卸载冲突包再安装兼容版本：

    !pip uninstall -y lightning pytorch-lightning
    !pip install pytorch-lightning==1.9.5
    

    安装完成后重启Colab运行时再执行代码。
• 本地文件命名冲突
  检查当前工作目录下是否存在命名为pytorch_lightning.py或lightning.py的自定义文件，该类文件会覆盖官方库的导入优先级，重命名文件即可解决问题。

内容的提问来源于stack exchange，提问作者Mercy Fas