进口pytorch-lightning，但是spyder好像找不到

Question

我需要

import LightningModule

从

pytorch_lightning

。 我没有

pytorch_lightning

，所以我安装了它，当我启动代码时，我有一个

ModuleNotFoundError

，告诉我有

No module named pytorch_lightning

.

第一个问题是：我安装的模块包括

LightningModule

，但被称为

pytorch-lightning

而不是

pytorch_lightning

，这会是问题的根源吗？ 如果不能，是否有人有解决方案的想法？ 提前致谢！

Answer 1

确保你的依赖安装在哪个python环境中。

您是否创建了专用的 python 虚拟环境？或者你的系统环境安装了吗？

下面的脚本按预期工作

# create virtualenvironment
python -m venv venv

# enable it
.\venv\Scripts\activate # WIN system
source ./venv/bin/activate # LINUX system

# install dependency 'pytorch_lightning'
python -m pip install pytorch_lightning

---

在 PYTHON REPL（或任何你想要的地方）中导入你的类

(venv) C:\Users\bigmauri\workspace\test>python
>>> import pytorch_lightning
>>> pytorch_lightning.LightningModule
<class 'pytorch_lightning.core.module.LightningModule'>

---

doc pythorc-lightning 这里

Answer 2

随着 PyTorch Lightning 2.0 的发展，正确的安装方式是 -

pip install lightning

import lightning as L

class Model(L.LightningModule):
    ...


trainer = L.Trainer()