报错记录

执行如下代码:

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的方法

  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
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐