报错记录
执行如下代码:
import pytorch_lightning
报错:
No module named ‘pytorch_lightning’
解决方式
安装pytorch_lightning包即可。
一般情况下,缺失的包通过pip安装,即:
pip install pytorch_lightning
然而,该指令默认安装最新版本的pytorch_lightning 2.1,在有些情况下,可能与本地环境不兼容。例如,我的Pytorch版本是1.10,但pytorch_lightning 2.1要求Pytorch版本 ≥1.12。
接下来介绍安装指定版本pytorch_lightning的方式。如果你不知道该安装哪一个pytorch_lightning版本,也可以通过阅读下文解决。
安装pytorch_lightning的方法
- 根据自己的CUDA环境,安装Pytorch。
具体可以参考:Previous PyTorch Versions | PyTorch
- 参照如下表格,根据自己的python与pytorch版本,确定需要安装的pytorch_lightning版本。
参考链接:Versioning Policy — PyTorch Lightning 2.1.3 documentation
例如,我的Python版本是3.8,我的Pytorch版本是1.10,因此我可以安装PyTorch Lightning 1.9。
- 安装兼容版本的PyTorch Lightning。
例如:
pip install pytorch_lightning==1.9