当前位置:   article > 正文

报错解决:No module named ‘pytorch_lightning‘ 安装pytorch_lightning_no module named 'pytorch_lightning

no module named 'pytorch_lightning

报错记录

执行如下代码:

import pytorch_lightning
  • 1

报错:
No module named ‘pytorch_lightning

解决方式

安装pytorch_lightning包即可。

一般情况下,缺失的包通过pip安装,即:

pip install pytorch_lightning
  • 1

然而,该指令默认安装最新版本的pytorch_lightning 2.1,在有些情况下,可能与本地环境不兼容。例如,我的Pytorch版本是1.10,但pytorch_lightning 2.1要求Pytorch版本 ≥1.12。

接下来介绍安装指定版本pytorch_lightning的方式。如果你不知道该安装哪一个pytorch_lightning版本,也可以通过阅读下文解决。

安装pytorch_lightning的方法

  1. 根据自己的CUDA环境,安装Pytorch。

具体可以参考:Previous PyTorch Versions | PyTorch

  1. 参照如下表格,根据自己的python与pytorch版本,确定需要安装的pytorch_lightning版本。

在这里插入图片描述
参考链接:Versioning Policy — PyTorch Lightning 2.1.3 documentation

例如,我的Python版本是3.8,我的Pytorch版本是1.10,因此我可以安装PyTorch Lightning 1.9。

  1. 安装兼容版本的PyTorch Lightning。

例如:

pip install pytorch_lightning==1.9
  • 1
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/472073
推荐阅读
相关标签
  

闽ICP备14008679号